Searched refs:start_index_shapes (Results 1 – 2 of 2) sorted by relevance
186 const Shape& operand_shape, absl::Span<const Shape> start_index_shapes,193 absl::Span<const Shape> start_index_shapes,
2266 const Shape& operand_shape, absl::Span<const Shape> start_index_shapes, in InferDynamicSliceShape() argument2269 auto number_of_indices = start_index_shapes.size(); in InferDynamicSliceShape()2272 (number_of_indices >= 1 && start_index_shapes[0].rank() == 1)) { in InferDynamicSliceShape()2279 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicSliceShape()2322 const Shape& first_index_shape = start_index_shapes[0]; in InferDynamicSliceShape()2331 for (const Shape& index_shape : start_index_shapes) { in InferDynamicSliceShape()2369 absl::Span<const Shape> start_index_shapes, bool allow_scalar_indices) { in InferDynamicUpdateSliceShape() argument2375 auto number_of_indices = start_index_shapes.size(); in InferDynamicUpdateSliceShape()2378 (number_of_indices >= 1 && start_index_shapes[0].rank() == 1)) { in InferDynamicUpdateSliceShape()2384 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicUpdateSliceShape()[all …]