Searched refs:allow_scalar_indices (Results 1 – 2 of 2) sorted by relevance
206 absl::Span<const int64> slice_sizes, bool allow_scalar_indices = true);213 bool allow_scalar_indices = true);
2475 absl::Span<const int64> slice_sizes, bool allow_scalar_indices) { in InferDynamicSliceShape() argument2479 if (!allow_scalar_indices || in InferDynamicSliceShape()2577 absl::Span<const Shape> start_index_shapes, bool allow_scalar_indices) { in InferDynamicUpdateSliceShape() argument2585 if (!allow_scalar_indices || in InferDynamicUpdateSliceShape()