Searched refs:N_big (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_nd_op.h | 67 int64 N_big = 1; in DoGatherNd() local
69 N_big *= indices_shape.dim_size(i); in DoGatherNd()
71 if (N_big > std::numeric_limits<int>::max()) { in DoGatherNd()
73 "indices has too many elements for int indexing: ", N_big, " > ", in DoGatherNd()
|
| D | scatter_op.cc | 99 const int64 N_big = indices.NumElements(); in DoCompute() local
101 c, N_big <= std::numeric_limits<Index>::max(), in DoCompute()
104 " indexing: ", N_big, " > ", in DoCompute()
|
| D | resource_variable_ops.cc | 888 const int64 N_big = indices.NumElements(); in DoCompute() local
890 c, N_big <= std::numeric_limits<Index>::max(), in DoCompute()
893 " indexing: ", N_big, " > ", in DoCompute()
895 const Index N = static_cast<Index>(N_big); in DoCompute()
|
| D | scatter_nd_op.cc | 847 const int64 N_big = indices.NumElements(); in PrepareAndValidateInputs() local
848 if (N_big > std::numeric_limits<Index>::max()) { in PrepareAndValidateInputs()
851 " indexing: ", N_big, " > ", in PrepareAndValidateInputs()
|