|
|
@@ -369,7 +369,7 @@ namespace Tensorflow |
|
|
|
var name_key = name.ToLower(); |
|
|
|
int i = 0; |
|
|
|
if (_names_in_use.ContainsKey(name_key)) |
|
|
|
i = _names_in_use[name_key];
|
|
|
|
i = _names_in_use[name_key]; |
|
|
|
// Increment the number for "name_key". |
|
|
|
if (mark_as_used) |
|
|
|
_names_in_use[name_key] = i + 1; |
|
|
@@ -399,13 +399,13 @@ namespace Tensorflow |
|
|
|
int num_return_outputs = 0; |
|
|
|
c_api.TF_ImportGraphDefResultsReturnOutputs(results, ref num_return_outputs, ref return_output_handle); |
|
|
|
TF_Output[] return_outputs = new TF_Output[num_return_outputs]; |
|
|
|
for (int i = 0; i < num_return_outputs; i++) |
|
|
|
unsafe |
|
|
|
{ |
|
|
|
var handle = return_output_handle + (Marshal.SizeOf<TF_Output>() * i); |
|
|
|
return_outputs[i] = Marshal.PtrToStructure<TF_Output>(handle); |
|
|
|
var tf_output_ptr = (TF_Output*) return_output_handle; |
|
|
|
for (int i = 0; i < num_return_outputs; i++) |
|
|
|
return_outputs[i] = *(tf_output_ptr + i); |
|
|
|
return return_outputs; |
|
|
|
} |
|
|
|
|
|
|
|
return return_outputs; |
|
|
|
} |
|
|
|
|
|
|
|
public string[] get_all_collection_keys() |
|
|
|