| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_conversion_folding.cc | 244 ShapeUtil::GetMutableSubshape(crs->mutable_shape(), {i}) in HandleAllReduce()
247 ShapeUtil::GetMutableSubshape(crs->mutable_shape(), {i})); in HandleAllReduce()
|
| D | memory_space_propagation.cc | 65 Shape* shape = ShapeUtil::GetMutableSubshape(instruction->mutable_shape(), in Propagate()
|
| D | bfloat16_propagation.cc | 930 ShapeUtil::GetMutableSubshape( in Run()
984 Shape* subshape = ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index); in OutputTypeAfterChange()
1007 entry.emplace(ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index), in AddToOrRemoveFromBF16ChangeSet()
1016 ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), index)); in AddToOrRemoveFromBF16ChangeSet()
|
| D | transfer_manager.cc | 219 *ShapeUtil::GetMutableSubshape(device_shape, index); in ReadDynamicShapes()
|
| D | bfloat16_normalization.cc | 351 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 | layout_assignment.cc | 1753 Shape* buffer_subshape = ShapeUtil::GetMutableSubshape( in AssignLayouts()
2036 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.cc | 1167 Shape* shape = ShapeUtil::GetMutableSubshape( in Finish()
2980 Shape* shape = ShapeUtil::GetMutableSubshape( in ExportAndColorBuffers()
|
| D | buffer_assignment.cc | 262 Shape* shape = ShapeUtil::GetMutableSubshape( in AddAssignment()
|
| D | pattern_matcher.h | 901 return ShapeUtil::GetMutableSubshape(shape, index_);
|
| D | memory_space_assignment_test.cc | 3143 *ShapeUtil::GetMutableSubshape(&tuple_shape, {0})->mutable_layout() = in TEST_P()
3148 *ShapeUtil::GetMutableSubshape(&tuple_shape, {1})->mutable_layout() = in TEST_P()
3153 *ShapeUtil::GetMutableSubshape(&tuple_shape, {2})->mutable_layout() = in TEST_P()
|
| D | hlo_instructions.cc | 1287 ShapeUtil::GetMutableSubshape(mutable_shape(), shape_index); in RelayoutConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_test.cc | 127 ShapeUtil::GetMutableSubshape(&dynamic_tuple, {2}) in TEST_F()
140 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 | 443 array_shape, *ShapeUtil::GetMutableSubshape(&array_shape, {}))); in TEST()
542 EXPECT_EQ(subshape, ShapeUtil::GetMutableSubshape(&shape, index)); in TEST()
|
| D | shape_util.h | 557 static Shape* GetMutableSubshape(Shape* shape, ShapeIndexView index);
|
| D | literal.cc | 535 Shape* subshape_ = ShapeUtil::GetMutableSubshape(shape_.get(), shape_index); in SetDynamicSize()
712 Shape* subshape = ShapeUtil::GetMutableSubshape(&new_shape, shape_index); in Relayout()
|
| D | shape_util.cc | 934 /* static */ Shape* ShapeUtil::GetMutableSubshape(Shape* shape, in GetMutableSubshape() function in xla::ShapeUtil
|
| /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/core/tpu/kernels/ |
| D | infeed_ops.cc | 137 xla::ShapeUtil::GetMutableSubshape(output_shape, {i}); in GetInfeedShapeWithLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 2954 *ShapeUtil::GetMutableSubshape(tuple->mutable_shape(), in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3186 *ShapeUtil::GetMutableSubshape(tuple->mutable_shape(), {2}) = in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|