Browse Source

Update Operation.Output.cs

tags/v0.9
Haiping GitHub 6 years ago
parent
commit
6bd08d6c4c
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/TensorFlowNET.Core/Operations/Operation.Output.cs

+ 2
- 2
src/TensorFlowNET.Core/Operations/Operation.Output.cs View File

@@ -42,8 +42,8 @@ namespace Tensorflow
if (NumControlOutputs > 0)
{
IntPtr control_output_handle = Marshal.AllocHGlobal(Marshal.SizeOf<IntPtr>() * NumControlOutputs);
c_api.TF_OperationGetControlOutputs(_handle, control_output_handle, NumControlInputs);
for (int i = 0; i < NumControlInputs; i++)
c_api.TF_OperationGetControlOutputs(_handle, control_output_handle, NumControlOutputs);
for (int i = 0; i < NumControlOutputs; i++)
{
var handle = control_output_handle + Marshal.SizeOf<IntPtr>() * i;
control_outputs[i] = new Operation(*(IntPtr*)handle);


Loading…
Cancel
Save