| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | computation_layout.cc | 34 SetToDefaultLayout(); in ComputationLayout()
40 void ComputationLayout::SetToDefaultLayout() { in SetToDefaultLayout() function in xla::ComputationLayout
42 parameter_layout.SetToDefaultLayout(); in SetToDefaultLayout()
44 result_layout_.SetToDefaultLayout(); in SetToDefaultLayout()
50 parameter_layout.SetToDefaultLayout(); in SetToDefaultLayoutIfEmpty()
54 result_layout_.SetToDefaultLayout(); in SetToDefaultLayoutIfEmpty()
|
| D | zero_sized_hlo_elimination.cc | 45 LayoutUtil::SetToDefaultLayout(&shape); in Run()
|
| D | computation_layout.h | 78 void SetToDefaultLayout();
|
| D | hlo_module_util.cc | 84 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
|
| D | while_util.cc | 232 LayoutUtil::SetToDefaultLayout(&shape); in MakeLoopStateShapeWithLayout()
|
| D | layout_assignment_test.cc | 1302 ->SetToDefaultLayout(); in AssignLayoutsToComputation()
|
| D | hlo_evaluator.cc | 1949 LayoutUtil::SetToDefaultLayout(instruction->mutable_shape()); in HandleFusion()
|
| D | hlo_parser.cc | 4604 LayoutUtil::SetToDefaultLayout(result); in ParseShape()
|
| D | algebraic_simplifier.cc | 3882 LayoutUtil::SetToDefaultLayout(&reshaped_shape); in HandleReshape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 229 LayoutUtil::SetToDefaultLayout(&op0_shape); in AddBackendConstraints()
231 LayoutUtil::SetToDefaultLayout(&op1_shape); in AddBackendConstraints()
233 LayoutUtil::SetToDefaultLayout(&output_shape); in AddBackendConstraints()
243 LayoutUtil::SetToDefaultLayout(&op0_shape); in AddBackendConstraints()
245 LayoutUtil::SetToDefaultLayout(&output_shape); in AddBackendConstraints()
284 LayoutUtil::SetToDefaultLayout(shape); in AddBackendConstraints()
|
| D | cusolver_rewriter.cc | 42 LayoutUtil::SetToDefaultLayout(shape); in SetFortranLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.cc | 144 /* static */ void LayoutUtil::SetToDefaultLayout(Shape* shape) { in SetToDefaultLayout() function in xla::LayoutUtil
148 SetToDefaultLayout(&element_shape); in SetToDefaultLayout()
164 LayoutUtil::SetToDefaultLayout(©); in GetWithDefaultLayout()
168 /* static */ void LayoutUtil::SetToDefaultLayout(ProgramShape* program_shape) { in SetToDefaultLayout() function in xla::LayoutUtil
170 LayoutUtil::SetToDefaultLayout(¶meter_shape); in SetToDefaultLayout()
172 LayoutUtil::SetToDefaultLayout(program_shape->mutable_result()); in SetToDefaultLayout()
|
| D | shape_layout.cc | 45 void ShapeLayout::SetToDefaultLayout() { in SetToDefaultLayout() function in xla::ShapeLayout
46 LayoutUtil::SetToDefaultLayout(&shape_); in SetToDefaultLayout()
|
| D | layout_util.h | 66 static void SetToDefaultLayout(Shape* shape);
74 static void SetToDefaultLayout(ProgramShape* program_shape);
|
| D | shape_layout.h | 59 void SetToDefaultLayout();
|
| D | layout_util_test.cc | 233 LayoutUtil::SetToDefaultLayout(&shape); in TEST_F()
|
| D | shape_util.cc | 355 LayoutUtil::SetToDefaultLayout(shape); in PopulateShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 121 LayoutUtil::SetToDefaultLayout(&result_shape); in CreateExecutionOptions()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | utils.cc | 178 LayoutUtil::SetToDefaultLayout(subshape); in DetermineArgumentLayoutsFromCompileOptions()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | outfeed_receiver.cc | 425 LayoutUtil::SetToDefaultLayout(subshape); in AddOutfeedToBuilder()
|
| D | xla.cc | 145 LayoutUtil::SetToDefaultLayout(subshape); in PYBIND11_MODULE()
|
| D | xla_compiler.cc | 240 LayoutUtil::SetToDefaultLayout(subshape); in BuildXlaCompilerSubmodule()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 96 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 479 xla::LayoutUtil::SetToDefaultLayout(&spine_shape); in MakeTuple()
|