Searched refs:GetMutableSubshape (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_test.cc | 124 ShapeUtil::GetMutableSubshape(&dynamic_tuple, {2}) in TEST_F()
137 ShapeUtil::GetMutableSubshape(&dynamic_tuple, {2}) in TEST_F()
|
| D | shape_layout.cc | 73 *ShapeUtil::GetMutableSubshape(&shape_, shape_index)->mutable_layout() = in ResetLayout()
|
| D | shape_util_test.cc | 419 array_shape, *ShapeUtil::GetMutableSubshape(&array_shape, {}))); in TEST()
518 EXPECT_EQ(subshape, ShapeUtil::GetMutableSubshape(&shape, index)); in TEST()
|
| D | shape_util.h | 495 static Shape* GetMutableSubshape(Shape* shape, ShapeIndexView index);
|
| D | shape_util.cc | 807 /* static */ Shape* ShapeUtil::GetMutableSubshape(Shape* shape, in GetMutableSubshape() function in xla::ShapeUtil
|
| D | literal.cc | 582 Shape* subshape = ShapeUtil::GetMutableSubshape(&new_shape, shape_index); in Relayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation.cc | 830 Shape* subshape = ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index); in OutputTypeAfterChange()
853 entry.emplace(ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index), in AddToOrRemoveFromBF16ChangeSet()
862 ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index)); in AddToOrRemoveFromBF16ChangeSet()
|
| D | bfloat16_conversion_folding.cc | 229 ShapeUtil::GetMutableSubshape(crs->mutable_shape(), {i}) in HandleAllReduce()
|
| D | bfloat16_normalization.cc | 239 auto subshape = ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), {i}); in HandleMultipleOutputs()
|
| D | layout_assignment_test.cc | 310 *ShapeUtil::GetMutableSubshape(&result_shape, /*index=*/{0, 0}) = in TEST_F()
312 *ShapeUtil::GetMutableSubshape(&result_shape, /*index=*/{1, 0}) = in TEST_F()
|
| D | hlo_instruction_test.cc | 742 *ShapeUtil::GetMutableSubshape(tuple->mutable_shape(), {0}) in TEST_F()
744 *ShapeUtil::GetMutableSubshape(tuple->mutable_shape(), {1}) in TEST_F()
|
| D | layout_assignment.cc | 1624 Shape* buffer_subshape = ShapeUtil::GetMutableSubshape( in AssignLayouts()
1857 ShapeUtil::GetMutableSubshape(instruction->mutable_shape(), {0}); in ConstrainChannelLayouts()
|
| D | pattern_matcher.h | 920 return ShapeUtil::GetMutableSubshape(shape, index_);
|
| D | hlo_instructions.cc | 1047 ShapeUtil::GetMutableSubshape(mutable_shape(), shape_index); in RelayoutConstant()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 364 *xla::ShapeUtil::GetMutableSubshape(host_shape, index) = in ExpandTreeOfTuples()
366 *xla::ShapeUtil::GetMutableSubshape(device_shape, index) = in ExpandTreeOfTuples()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 77 *xla::ShapeUtil::GetMutableSubshape(shape, {}) = in ParseTupleNode()
82 xla::ShapeUtil::GetMutableSubshape(shape, {i}), rm)); in ParseTupleNode()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 333 xla::Shape* output_sub_shape = xla::ShapeUtil::GetMutableSubshape( in BuildComputation()
|