Home
last modified time | relevance | path

Searched refs:input_buffer_shape (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_verifier.cc569 const Shape& input_buffer_shape = collective_permute->operand(0)->shape(); in CheckInplaceCollectivePermute() local
574 if (input_buffer_shape.IsArray() && output_buffer_shape.IsArray()) { in CheckInplaceCollectivePermute()
576 CheckBufferOffset(input_buffer_shape, input_offset_shape); in CheckInplaceCollectivePermute()
585 } else if (input_buffer_shape.IsTuple() && output_buffer_shape.IsTuple()) { in CheckInplaceCollectivePermute()
586 if (ShapeUtil::TupleElementCount(input_buffer_shape) != in CheckInplaceCollectivePermute()
592 ShapeUtil::TupleElementCount(input_buffer_shape)) { in CheckInplaceCollectivePermute()
595 for (int i = 0; i < input_buffer_shape.tuple_shapes_size(); ++i) { in CheckInplaceCollectivePermute()
597 CheckBufferOffset(input_buffer_shape.tuple_shapes(i), in CheckInplaceCollectivePermute()