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