Searched refs:N_big (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op.cc | 105 const int64 N_big = indices.NumElements(); in DoCompute() local
107 c, N_big <= std::numeric_limits<Index>::max(), in DoCompute()
110 " indexing: ", N_big, " > ", in DoCompute()
184 const int64 N_big = indices.NumElements(); in DoCompute() local
186 c, N_big <= std::numeric_limits<Index>::max(), in DoCompute()
189 " indexing: ", N_big, " > ", in DoCompute()
|
| D | gather_nd_op.cc | 100 int64 N_big = 1; in DoGatherNd() local
102 N_big *= indices_shape.dim_size(i); in DoGatherNd()
104 if (N_big > std::numeric_limits<int>::max()) { in DoGatherNd()
106 "indices has too many elements for int indexing: ", N_big, " > ", in DoGatherNd()
|
| D | resource_variable_ops.cc | 769 const int64 N_big = indices.NumElements(); in Compute() local
771 c, N_big <= std::numeric_limits<Index>::max(), in Compute()
774 " indexing: ", N_big, " > ", in Compute()
776 const Index N = static_cast<Index>(N_big); in Compute()
|
| D | scatter_nd_op.cc | 583 const int64 N_big = indices.NumElements(); in PrepareAndValidateInputs() local
584 if (N_big > std::numeric_limits<Index>::max()) { in PrepareAndValidateInputs()
587 " indexing: ", N_big, " > ", in PrepareAndValidateInputs()
|