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