Home
last modified time | relevance | path

Searched refs:SetToDefaultLayout (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.cc33 SetToDefaultLayout(); in ComputationLayout()
37 void ComputationLayout::SetToDefaultLayout() { in SetToDefaultLayout() function in xla::ComputationLayout
39 parameter_layout.SetToDefaultLayout(); in SetToDefaultLayout()
41 result_layout_.SetToDefaultLayout(); in SetToDefaultLayout()
Dzero_sized_hlo_elimination.cc44 LayoutUtil::SetToDefaultLayout(&shape); in Run()
Dcomputation_layout.h78 void SetToDefaultLayout();
Dlocal_service.cc109 LayoutUtil::SetToDefaultLayout(&result_shape); in CreateExecutionOptions()
Dservice.cc302 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
Dlayout_assignment_test.cc1308 ->SetToDefaultLayout(); in AssignLayoutsToComputation()
Dhlo_evaluator.cc1237 LayoutUtil::SetToDefaultLayout(instruction->mutable_shape()); in HandleFusion()
Dalgebraic_simplifier.cc2598 LayoutUtil::SetToDefaultLayout(&reshaped_shape); in HandleReshape()
Dhlo_parser.cc3561 LayoutUtil::SetToDefaultLayout(result); in ParseShape()
/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util.cc142 /* static */ void LayoutUtil::SetToDefaultLayout(Shape* shape) { in SetToDefaultLayout() function in xla::LayoutUtil
146 SetToDefaultLayout(&element_shape); in SetToDefaultLayout()
162 LayoutUtil::SetToDefaultLayout(&copy); in GetWithDefaultLayout()
166 /* static */ void LayoutUtil::SetToDefaultLayout(ProgramShape* program_shape) { in SetToDefaultLayout() function in xla::LayoutUtil
168 LayoutUtil::SetToDefaultLayout(&parameter_shape); in SetToDefaultLayout()
170 LayoutUtil::SetToDefaultLayout(program_shape->mutable_result()); in SetToDefaultLayout()
Dshape_layout.cc45 void ShapeLayout::SetToDefaultLayout() { in SetToDefaultLayout() function in xla::ShapeLayout
46 LayoutUtil::SetToDefaultLayout(&shape_); in SetToDefaultLayout()
Dlayout_util.h65 static void SetToDefaultLayout(Shape* shape);
73 static void SetToDefaultLayout(ProgramShape* program_shape);
Dshape_layout.h58 void SetToDefaultLayout();
Dlayout_util_test.cc299 LayoutUtil::SetToDefaultLayout(&shape); in TEST_F()
Dshape_util.cc233 LayoutUtil::SetToDefaultLayout(shape); in PopulateShape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc206 LayoutUtil::SetToDefaultLayout(&op0_shape); in AddBackendConstraints()
208 LayoutUtil::SetToDefaultLayout(&op1_shape); in AddBackendConstraints()
210 LayoutUtil::SetToDefaultLayout(&output_shape); in AddBackendConstraints()
251 LayoutUtil::SetToDefaultLayout(shape); in AddBackendConstraints()
Dcusolver_rewriter.cc43 LayoutUtil::SetToDefaultLayout(shape); in SetFortranLayout()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxrt.cc160 LayoutUtil::SetToDefaultLayout(&program_shape); in CompileForXrt()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc405 xla::LayoutUtil::SetToDefaultLayout(&spine_shape); in MakeTuple()