Browse Source

Add TF_UpdateEdge APIs.

v0.20-tensorflow2.3
Oceania2018 Haiping 5 years ago
parent
commit
375fe28f57
4 changed files with 3 additions and 7 deletions
  1. +1
    -1
      src/TensorFlowNET.Core/APIs/c_api.cs
  2. +0
    -4
      src/TensorFlowNET.Core/Graphs/Graph.cs
  3. +1
    -1
      src/TensorFlowNET.Core/Graphs/c_api.graph.cs
  4. +1
    -1
      src/TensorFlowNET.Core/Operations/Operation.cs

+ 1
- 1
src/TensorFlowNET.Core/APIs/c_api.cs View File

@@ -43,7 +43,7 @@ namespace Tensorflow
/// </summary> /// </summary>
public partial class c_api public partial class c_api
{ {
public const string TensorFlowLibName = @"C:\Users\haipi\Documents\Projects\tensorflow\bazel-bin\tensorflow\tensorflow";
public const string TensorFlowLibName = "tensorflow";


public static string StringPiece(IntPtr handle) public static string StringPiece(IntPtr handle)
{ {


+ 0
- 4
src/TensorFlowNET.Core/Graphs/Graph.cs View File

@@ -278,11 +278,7 @@ namespace Tensorflow
// after removing the trailing '/'. // after removing the trailing '/'.
name = name.EndsWith("/") ? ops.name_from_scope_name(name) : unique_name(name); name = name.EndsWith("/") ? ops.name_from_scope_name(name) : unique_name(name);
var node_def = ops._NodeDef(op_type, name, attrs: attrs); var node_def = ops._NodeDef(op_type, name, attrs: attrs);
if (name == "rnn/while/basic_rnn_cell/MatMul"
|| name == "rnn/while/basic_rnn_cell/MatMul/Enter")
{


}
var input_ops = inputs.Select(x => x.op).ToArray(); var input_ops = inputs.Select(x => x.op).ToArray();
var control_inputs = _control_dependencies_for_inputs(input_ops); var control_inputs = _control_dependencies_for_inputs(input_ops);




+ 1
- 1
src/TensorFlowNET.Core/Graphs/c_api.graph.cs View File

@@ -321,6 +321,6 @@ namespace Tensorflow
/// <param name="status">TF_Status*</param> /// <param name="status">TF_Status*</param>
[DllImport(TensorFlowLibName)] [DllImport(TensorFlowLibName)]
public static extern void UpdateEdge(IntPtr graph, TF_Output new_src, TF_Input dst, SafeStatusHandle status);
public static extern void TF_UpdateEdge(IntPtr graph, TF_Output new_src, TF_Input dst, SafeStatusHandle status);
} }
} }

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

@@ -326,7 +326,7 @@ namespace Tensorflow
// the updated inputs are reloaded from the c_api // the updated inputs are reloaded from the c_api
lock (Locks.ProcessWide) lock (Locks.ProcessWide)
{ {
c_api.UpdateEdge(_graph, output, input, tf.Status.Handle);
c_api.TF_UpdateEdge(_graph, output, input, tf.Status.Handle);
//var updated_inputs = inputs; //var updated_inputs = inputs;
tf.Status.Check(); tf.Status.Check();
} }


Loading…
Cancel
Save