| /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 | 238 LayoutUtil::SetToDefaultLayout(&shape); in MakeLoopStateShapeWithLayout()
|
| D | layout_assignment_test.cc | 1330 ->SetToDefaultLayout(); in AssignLayoutsToComputation()
|
| D | hlo_evaluator.cc | 2013 LayoutUtil::SetToDefaultLayout(instruction->mutable_shape()); in HandleFusion()
|
| D | hlo_parser.cc | 4963 LayoutUtil::SetToDefaultLayout(result); in ParseShape()
|
| D | algebraic_simplifier.cc | 3973 LayoutUtil::SetToDefaultLayout(&reshaped_shape); in HandleReshape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 248 LayoutUtil::SetToDefaultLayout(&op0_shape); in AddBackendConstraints()
250 LayoutUtil::SetToDefaultLayout(&op1_shape); in AddBackendConstraints()
252 LayoutUtil::SetToDefaultLayout(&output_shape); in AddBackendConstraints()
262 LayoutUtil::SetToDefaultLayout(&op0_shape); in AddBackendConstraints()
264 LayoutUtil::SetToDefaultLayout(&output_shape); in AddBackendConstraints()
303 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 | layout_util.h | 66 static void SetToDefaultLayout(Shape* shape);
74 static void SetToDefaultLayout(ProgramShape* program_shape);
|
| D | shape_layout.cc | 45 void ShapeLayout::SetToDefaultLayout() { in SetToDefaultLayout() function in xla::ShapeLayout
46 LayoutUtil::SetToDefaultLayout(&shape_); in SetToDefaultLayout()
|
| D | shape_layout.h | 62 void SetToDefaultLayout();
|
| D | layout_util_test.cc | 233 LayoutUtil::SetToDefaultLayout(&shape); in TEST_F()
|
| D | shape_util.cc | 334 LayoutUtil::SetToDefaultLayout(&ret); in MoveDimToMajor()
383 LayoutUtil::SetToDefaultLayout(shape); in PopulateShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 115 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 | 433 LayoutUtil::SetToDefaultLayout(subshape); in AddOutfeedToBuilder()
|
| D | xla.cc | 149 LayoutUtil::SetToDefaultLayout(subshape); in PYBIND11_MODULE()
|
| D | xla_compiler.cc | 290 LayoutUtil::SetToDefaultLayout(subshape); in BuildXlaCompilerSubmodule()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 98 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()
|