@@ -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) | ||||
{ | { | ||||
@@ -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); | ||||
@@ -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); | |||||
} | } | ||||
} | } |
@@ -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(); | ||||
} | } | ||||