| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_conversion_folding.cc | 238 ShapeUtil::GetMutableSubshape(crs->mutable_shape(), {i}) in HandleAllReduce()
241 ShapeUtil::GetMutableSubshape(crs->mutable_shape(), {i})); in HandleAllReduce()
|
| D | bfloat16_propagation.cc | 828 ShapeUtil::GetMutableSubshape( in Run()
882 Shape* subshape = ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index); in OutputTypeAfterChange()
905 entry.emplace(ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index), in AddToOrRemoveFromBF16ChangeSet()
914 ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index)); in AddToOrRemoveFromBF16ChangeSet()
|
| D | bfloat16_normalization.cc | 351 auto subshape = ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), {i}); in HandleMultipleOutputs()
|
| D | memory_space_assignment.cc | 353 Shape* shape = ShapeUtil::GetMutableSubshape( in Finish()
1398 Shape* shape = ShapeUtil::GetMutableSubshape( in Process()
|
| 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 | layout_assignment.cc | 1730 Shape* buffer_subshape = ShapeUtil::GetMutableSubshape( in AssignLayouts()
1996 Shape* shape = ShapeUtil::GetMutableSubshape( in PropagateMemorySpace()
|
| D | hlo_instruction_test.cc | 744 *ShapeUtil::GetMutableSubshape(tuple->mutable_shape(), {0}) in TEST_F()
746 *ShapeUtil::GetMutableSubshape(tuple->mutable_shape(), {1}) in TEST_F()
|
| D | memory_space_assignment_test.cc | 2026 *ShapeUtil::GetMutableSubshape(&tuple_shape, {0})->mutable_layout() = in TEST_P()
2032 *ShapeUtil::GetMutableSubshape(&tuple_shape, {1})->mutable_layout() = in TEST_P()
2036 *ShapeUtil::GetMutableSubshape(&tuple_shape, {2})->mutable_layout() = in TEST_P()
|
| D | buffer_assignment.cc | 258 Shape* shape = ShapeUtil::GetMutableSubshape( in AddAssignment()
|
| D | pattern_matcher.h | 914 return ShapeUtil::GetMutableSubshape(shape, index_);
|
| D | hlo_instructions.cc | 1146 ShapeUtil::GetMutableSubshape(mutable_shape(), shape_index); in RelayoutConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_test.cc | 126 ShapeUtil::GetMutableSubshape(&dynamic_tuple, {2}) in TEST_F()
139 ShapeUtil::GetMutableSubshape(&dynamic_tuple, {2}) in TEST_F()
|
| D | shape_layout.cc | 78 *ShapeUtil::GetMutableSubshape(&shape_, shape_index)->mutable_layout() = in ResetLayout()
|
| D | shape_util_test.cc | 441 array_shape, *ShapeUtil::GetMutableSubshape(&array_shape, {}))); in TEST()
540 EXPECT_EQ(subshape, ShapeUtil::GetMutableSubshape(&shape, index)); in TEST()
|
| D | shape_util.h | 508 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 | 576 Shape* subshape = ShapeUtil::GetMutableSubshape(&new_shape, shape_index); in Relayout()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 438 *xla::ShapeUtil::GetMutableSubshape(host_shape, index) = in ExpandTreeOfTuples()
440 *xla::ShapeUtil::GetMutableSubshape(device_shape, index) = in ExpandTreeOfTuples()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 81 *xla::ShapeUtil::GetMutableSubshape(shape, {}) = in ParseTupleNode()
86 xla::ShapeUtil::GetMutableSubshape(shape, {i}), rm)); in ParseTupleNode()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 407 xla::Shape* output_sub_shape = xla::ShapeUtil::GetMutableSubshape( in BuildComputation()
|