Skip to content

Commit 454a55b

Browse files
NiklasGustafssonEsther2013
authored andcommitted
Corrected fast path call to GatherV2
1 parent 37ca023 commit 454a55b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/TensorFlowNET.Core/Operations/gen_array_ops.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ public static Tensor gather_v2<T1, T2>(T1 @params, T2 indices, int axis, string
123123
{
124124
try
125125
{
126-
var results = tf.Runner.TFE_FastPathExecute(new FastPathOpExecInfo("GatherV2", name, @params, indices, axis, "batch_dims")
126+
var results = tf.Runner.TFE_FastPathExecute(new FastPathOpExecInfo("GatherV2", name, @params, indices, axis, "batch_dims", 0)
127127
{
128128
ctx = tf.Context,
129129
device_name = tf.Context.DeviceName

0 commit comments

Comments
 (0)