From 454a55b35c39ab2fe6f7362b0c45777572542908 Mon Sep 17 00:00:00 2001 From: Niklas Gustafsson Date: Tue, 9 Mar 2021 16:19:22 -0800 Subject: [PATCH] Corrected fast path call to GatherV2 --- src/TensorFlowNET.Core/Operations/gen_array_ops.cs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/TensorFlowNET.Core/Operations/gen_array_ops.cs b/src/TensorFlowNET.Core/Operations/gen_array_ops.cs index 4482e979..5dbbb3f9 100644 --- a/src/TensorFlowNET.Core/Operations/gen_array_ops.cs +++ b/src/TensorFlowNET.Core/Operations/gen_array_ops.cs @@ -123,7 +123,7 @@ namespace Tensorflow { try { - var results = tf.Runner.TFE_FastPathExecute(new FastPathOpExecInfo("GatherV2", name, @params, indices, axis, "batch_dims") + var results = tf.Runner.TFE_FastPathExecute(new FastPathOpExecInfo("GatherV2", name, @params, indices, axis, "batch_dims", 0) { ctx = tf.Context, device_name = tf.Context.DeviceName