Home
last modified time | relevance | path

Searched refs:mutable_result_layout (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc115 *computation_layout.mutable_result_layout() = shape_layout; in TEST_F()
150 *computation_layout.mutable_result_layout() = col_major; in TEST_F()
195 *computation_layout.mutable_result_layout() = shape_layout; in TEST_F()
280 TF_CHECK_OK(computation_layout.mutable_result_layout()->CopyLayoutFromShape( in TEST_F()
340 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout); in TEST_F()
379 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout); in TEST_F()
417 *computation_layout.mutable_result_layout() = in TEST_F()
468 *computation_layout.mutable_result_layout() = in TEST_F()
692 computation_layout.mutable_result_layout()->ResetLayout( in TEST_F()
859 computation_layout.mutable_result_layout()->ResetLayout( in TEST_F()
[all …]
Dhlo_module_util.cc80 computation_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfig()
84 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
177 *module->mutable_entry_computation_layout()->mutable_result_layout() = in UpdateEntryComputationLayout()
Dcomputation_layout.h68 ShapeLayout* mutable_result_layout() { return &result_layout_; } in mutable_result_layout() function
Dlayout_assignment.h165 *computation_layout_.mutable_result_layout() = shape_layout; in ResetResultLayout()
Dlayout_assignment.cc771 *branch_computation_layout.mutable_result_layout() = in AddMandatoryConstraints()
2113 UpdateLayout(result, callee_layout.mutable_result_layout())) { in CalculateComputationLayout()
2398 ShapeLayout* result_layout = computation_layout->mutable_result_layout(); in PropagateComputationLayouts()
Dhlo_module.cc578 TF_RETURN_IF_ERROR(entry_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfigFromShape()
Ddynamic_padder_test.cc540 ->mutable_result_layout() in PadAndExecute()
Dhlo_parser.cc814 *computation_layout->mutable_result_layout() = ShapeLayout(result); in ParseComputationLayout()
1007 ->mutable_result_layout() in ParseComputations()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc81 *computation_layout.mutable_result_layout() = in TEST_F()
122 *computation_layout.mutable_result_layout() = in TEST_F()
168 *computation_layout.mutable_result_layout() = in TEST_F()
200 *computation_layout.mutable_result_layout() = in TEST_F()
240 *computation_layout.mutable_result_layout() = in TEST_F()
317 *computation_layout.mutable_result_layout() = in RunDotOutputFusion()
461 *computation_layout.mutable_result_layout() = in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.h313 ->mutable_result_layout() in ForceResultLayout()
320 ->mutable_result_layout() in ForceResultLayout()
328 ->mutable_result_layout() in ForceClearResultLayout()
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.cc110 *output.mutable_result_layout() = result_layout()->ToProto(); in ToProto()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc94 computation_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfig()
98 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc79 *computation_layout.mutable_result_layout() = in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1742 *bitcast_config.mutable_result_layout() = result_layout; in ExportXlaOp()