Commit f41d6bc
committed
Add TF_UpdateEdge APIs.
1 parent 5f2df56 commit f41d6bc
4 files changed
Lines changed: 3 additions & 7 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -43,7 +43,7 @@ namespace Tensorflow | |||
| 43 | 43 | /// </summary> | |
| 44 | 44 | public partial class c_api | |
| 45 | 45 | { | |
| 46 | - public const string TensorFlowLibName = @"C:\Users\haipi\Documents\Projects\tensorflow\bazel-bin\tensorflow\tensorflow"; | ||
| 46 | + public const string TensorFlowLibName = "tensorflow"; | ||
| 47 | 47 | ||
| 48 | 48 | public static string StringPiece(IntPtr handle) | |
| 49 | 49 | { | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -278,11 +278,7 @@ public Operation create_op(string op_type, Tensor[] inputs, TF_DataType[] dtypes | |||
| 278 | 278 | // after removing the trailing '/'. | |
| 279 | 279 | name = name.EndsWith("/") ? ops.name_from_scope_name(name) : unique_name(name); | |
| 280 | 280 | var node_def = ops._NodeDef(op_type, name, attrs: attrs); | |
| 281 | - if (name == "rnn/while/basic_rnn_cell/MatMul" | ||
| 282 | - || name == "rnn/while/basic_rnn_cell/MatMul/Enter") | ||
| 283 | - { | ||
| 284 | 281 | ||
| 285 | - } | ||
| 286 | 282 | var input_ops = inputs.Select(x => x.op).ToArray(); | |
| 287 | 283 | var control_inputs = _control_dependencies_for_inputs(input_ops); | |
| 288 | 284 | ||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -321,6 +321,6 @@ public static extern void TF_GraphSetOutputHandleShapesAndTypes(IntPtr graph, TF | |||
| 321 | 321 | /// <param name="status">TF_Status*</param> | |
| 322 | 322 | [DllImport(TensorFlowLibName)] | |
| 323 | 323 | ||
| 324 | - public static extern void UpdateEdge(IntPtr graph, TF_Output new_src, TF_Input dst, SafeStatusHandle status); | ||
| 324 | + public static extern void TF_UpdateEdge(IntPtr graph, TF_Output new_src, TF_Input dst, SafeStatusHandle status); | ||
| 325 | 325 | } | |
| 326 | 326 | } | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -326,7 +326,7 @@ public void _update_input(int index, Tensor tensor) | |||
| 326 | 326 | // the updated inputs are reloaded from the c_api | |
| 327 | 327 | lock (Locks.ProcessWide) | |
| 328 | 328 | { | |
| 329 | - c_api.UpdateEdge(_graph, output, input, tf.Status.Handle); | ||
| 329 | + c_api.TF_UpdateEdge(_graph, output, input, tf.Status.Handle); | ||
| 330 | 330 | //var updated_inputs = inputs; | |
| 331 | 331 | tf.Status.Check(); | |
| 332 | 332 | } | |
0 commit comments