Searched refs:allow_scalar_indices (Results 1 – 2 of 2) sorted by relevance
187 absl::Span<const int64> slice_sizes, bool allow_scalar_indices = true);194 bool allow_scalar_indices = true);
2267 absl::Span<const int64> slice_sizes, bool allow_scalar_indices) { in InferDynamicSliceShape() argument2271 if (!allow_scalar_indices || in InferDynamicSliceShape()2369 absl::Span<const Shape> start_index_shapes, bool allow_scalar_indices) { in InferDynamicUpdateSliceShape() argument2377 if (!allow_scalar_indices || in InferDynamicUpdateSliceShape()