Searched refs:slice_size_big (Results 1 – 2 of 2) sorted by relevance
96 int64 slice_size_big = 1; in DoGatherNd() local98 slice_size_big *= params_shape.dim_size(i); in DoGatherNd()102 if (slice_size_big > std::numeric_limits<Index>::max()) { in DoGatherNd()104 "slice size is too large for indexing: ", slice_size_big, " > ", in DoGatherNd()108 const Index slice_size = static_cast<Index>(slice_size_big); in DoGatherNd()
871 int64 slice_size_big = 1; in PrepareAndValidateInputs() local873 slice_size_big *= params_shape.dim_size(i); in PrepareAndValidateInputs()876 if (slice_size_big > std::numeric_limits<Index>::max()) { in PrepareAndValidateInputs()878 "slice size is too large for indexing: ", slice_size_big, " > ", in PrepareAndValidateInputs()882 *slice_size = static_cast<Index>(slice_size_big); in PrepareAndValidateInputs()