Searched refs:ref_tensor_shape (Results 1 – 1 of 1) sorted by relevance
2054 auto ref_tensor_shape = MaybeGetMinimumShape( in PredictScatter() local2057 for (int i = 1; i < ref_tensor_shape.dim().size(); ++i) { in PredictScatter()2058 num_elems_in_ref_per_index *= ref_tensor_shape.dim(i).size(); in PredictScatter()