HomeSort by relevance Sort by last modified time
    Searched refs:gather_op (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/python/kernel_tests/
gather_nd_op_test.py 336 gather_op = array_ops.gather_nd(t_params, t_indices)
339 gather_op.eval()
342 gather_op.eval()
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 1038 auto* gather_op = tensorflow_graph->add_node(); local
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
reduction_gpu_kernels.cu.h 673 GatherOp gather_op(extent_x, extent_y, extent_z, false);
677 gatherIterType gather_iter(counting_iter, gather_op);
    [all...]

Completed in 79 milliseconds