Searched refs:input_buffer_shape (Results 1 – 1 of 1) sorted by relevance
569 const Shape& input_buffer_shape = collective_permute->operand(0)->shape(); in CheckInplaceCollectivePermute() local574 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()