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

  /external/tensorflow/tensorflow/core/kernels/
gather_nd_op.cc 124 Index total_nd = params_shape.dims(); local
130 for (Index i = indices_nd; i < total_nd; ++i) {
scatter_nd_op.cc 380 Index total_nd = params_shape.dims(); local
383 for (int64 i = *slice_dim; i < total_nd; ++i) {
  /external/tensorflow/tensorflow/compiler/tests/
scatter_nd_op_test.py 52 total_nd = len(ref.shape)
54 for i in range(ixdim, total_nd):
  /external/tensorflow/tensorflow/python/kernel_tests/
scatter_nd_ops_test.py 57 total_nd = len(ref.shape)
59 for i in range(ixdim, total_nd):

Completed in 198 milliseconds