Searched refs:start_index_shapes (Results 1 – 2 of 2) sorted by relevance
205 const Shape& operand_shape, absl::Span<const Shape> start_index_shapes,212 absl::Span<const Shape> start_index_shapes,
2474 const Shape& operand_shape, absl::Span<const Shape> start_index_shapes, in InferDynamicSliceShape() argument2477 auto number_of_indices = start_index_shapes.size(); in InferDynamicSliceShape()2480 (number_of_indices >= 1 && start_index_shapes[0].rank() == 1)) { in InferDynamicSliceShape()2487 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicSliceShape()2530 const Shape& first_index_shape = start_index_shapes[0]; in InferDynamicSliceShape()2539 for (const Shape& index_shape : start_index_shapes) { in InferDynamicSliceShape()2577 absl::Span<const Shape> start_index_shapes, bool allow_scalar_indices) { in InferDynamicUpdateSliceShape() argument2583 auto number_of_indices = start_index_shapes.size(); in InferDynamicUpdateSliceShape()2586 (number_of_indices >= 1 && start_index_shapes[0].rank() == 1)) { in InferDynamicUpdateSliceShape()2592 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicUpdateSliceShape()[all …]