Searched refs:num_elems_in_ref_per_index (Results 1 – 1 of 1) sorted by relevance
2053 int64_t num_elems_in_ref_per_index = 1; in PredictScatter() local2058 num_elems_in_ref_per_index *= ref_tensor_shape.dim(i).size(); in PredictScatter()2060 const int64_t op_count = num_indices * num_elems_in_ref_per_index; in PredictScatter()