| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A 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()
|
| H A D | zero_sized_hlo_elimination.cc | 48 LayoutUtil::SetToDefaultLayout(&shape); in Run()
|
| H A D | computation_layout.h | 78 void SetToDefaultLayout();
|
| H A D | hlo_module_util.cc | 84 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
|
| H A D | while_util.cc | 238 LayoutUtil::SetToDefaultLayout(&shape); in MakeLoopStateShapeWithLayout()
|
| H A D | layout_assignment_test.cc | 1338 ->SetToDefaultLayout(); in AssignLayoutsToComputation()
|
| H A D | hlo_evaluator.cc | 3154 LayoutUtil::SetToDefaultLayout(instruction->mutable_shape()); in HandleFusion()
|
| H A D | hlo_parser.cc | 5223 LayoutUtil::SetToDefaultLayout(result); in ParseShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | layout_util.cc | 145 /* static */ void LayoutUtil::SetToDefaultLayout(Shape* shape) { in SetToDefaultLayout() function in xla::LayoutUtil
149 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()
|
| H A D | shape_layout.cc | 45 void ShapeLayout::SetToDefaultLayout() { in SetToDefaultLayout() function in xla::ShapeLayout
46 LayoutUtil::SetToDefaultLayout(&shape_); in SetToDefaultLayout()
|
| H A D | layout_util.h | 66 static void SetToDefaultLayout(Shape* shape);
74 static void SetToDefaultLayout(ProgramShape* program_shape);
|
| H A D | shape_layout.h | 62 void SetToDefaultLayout();
|
| H A D | layout_util_test.cc | 275 LayoutUtil::SetToDefaultLayout(&shape); in TEST_F()
|
| H A D | shape_util.cc | 337 LayoutUtil::SetToDefaultLayout(&ret); in MoveDimToMajor()
392 LayoutUtil::SetToDefaultLayout(shape); in PopulateShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_layout_assignment.cc | 223 LayoutUtil::SetToDefaultLayout(shape); in SetFortranLayout()
330 LayoutUtil::SetToDefaultLayout(&op0_shape); in AddBackendConstraints()
332 LayoutUtil::SetToDefaultLayout(&output_shape); in AddBackendConstraints()
414 LayoutUtil::SetToDefaultLayout(&shape); in SetDotOperandLayout()
|
| H A D | cusolver_rewriter.cc | 44 LayoutUtil::SetToDefaultLayout(shape); in SetFortranLayout()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | executable_build_options.cc | 191 LayoutUtil::SetToDefaultLayout(&result_shape); in CreateExecutionOptions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | utils.cc | 180 LayoutUtil::SetToDefaultLayout(subshape); in DetermineArgumentLayoutsFromCompileOptions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | outfeed_receiver.cc | 432 LayoutUtil::SetToDefaultLayout(subshape); in AddOutfeedToBuilder()
|
| H A D | xla.cc | 165 LayoutUtil::SetToDefaultLayout(subshape); in PYBIND11_MODULE()
|
| H A D | xla_compiler.cc | 347 LayoutUtil::SetToDefaultLayout(subshape); in BuildXlaCompilerSubmodule()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | light_outside_compilation.cc | 133 xla::LayoutUtil::SetToDefaultLayout(&operand_shapes_with_layout.back()); in CompileToCustomCallCallingTfKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_support.cc | 98 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/ |
| H A D | hlo_proto_to_memory_visualization_utils.cc | 149 LayoutUtil::SetToDefaultLayout(&shape); in UnpaddedSize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | xrt_state.cc | 479 xla::LayoutUtil::SetToDefaultLayout(&spine_shape); in MakeTuple()
|