Searched refs:num_elems_in_ref_per_index (Results 1 – 1 of 1) sorted by relevance
2019 int64 num_elems_in_ref_per_index = 1; in PredictScatter() local2024 num_elems_in_ref_per_index *= ref_tensor_shape.dim(i).size(); in PredictScatter()2026 const int64 op_count = num_indices * num_elems_in_ref_per_index; in PredictScatter()