Browse Source

Corrected fast path call to GatherV2

tags/v0.40-tf2.4-tstring
Niklas Gustafsson Esther Hu 4 years ago
parent
commit
454a55b35c
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/TensorFlowNET.Core/Operations/gen_array_ops.cs

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

@@ -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


Loading…
Cancel
Save