Searched refs:GetMutableSubshape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_normalization.cc | 196 auto subshape = ShapeUtil::GetMutableSubshape(crs->mutable_shape(), {i}); in HandleCrossReplicaSum()
|
| D | layout_assignment_test.cc | 281 *ShapeUtil::GetMutableSubshape(&result_shape, /*index=*/{0, 0}) = in TEST_F()
283 *ShapeUtil::GetMutableSubshape(&result_shape, /*index=*/{1, 0}) = in TEST_F()
|
| D | hlo_instruction_test.cc | 738 *ShapeUtil::GetMutableSubshape(tuple->mutable_shape(), {0}) in TEST_F()
740 *ShapeUtil::GetMutableSubshape(tuple->mutable_shape(), {1}) in TEST_F()
|
| D | layout_assignment.cc | 1397 Shape* buffer_subshape = ShapeUtil::GetMutableSubshape( in AssignLayouts()
|
| D | hlo_instruction.cc | 3410 ShapeUtil::GetMutableSubshape(mutable_shape(), shape_index); in RelayoutConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 382 array_shape, *ShapeUtil::GetMutableSubshape(&array_shape, {}))); in TEST()
536 EXPECT_EQ(subshape, ShapeUtil::GetMutableSubshape(&shape, index)); in TEST()
|
| D | shape_util.h | 436 static Shape* GetMutableSubshape(Shape* shape, ShapeIndexView index);
|
| D | shape_util.cc | 827 /* static */ Shape* ShapeUtil::GetMutableSubshape(Shape* shape, in GetMutableSubshape() function in xla::ShapeUtil
|
| D | literal_util.cc | 709 Shape* subshape = ShapeUtil::GetMutableSubshape(&new_shape, shape_index); in Relayout()
|