Searched defs:indices_nd (Results 1 – 4 of 4) sorted by relevance
64 const int64_t indices_nd = indices_shape.dim_size(indices_shape.dims() - 1); in DoGatherNd() local
73 const int indices_nd = SizeOfDimension(indices, indices_rank - 1); in Prepare() local
567 int indices_nd; member670 const int indices_nd = indices_shape.Dims(outer_dims); in ScatterNd() local
1452 const int indices_nd = in ProcessGatherNdOperator() local