Home
last modified time | relevance | path

Searched refs:mutable_parameter_layout (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment_test.cc113 *computation_layout.mutable_parameter_layout(0) = shape_layout; in TEST_F()
114 *computation_layout.mutable_parameter_layout(1) = shape_layout; in TEST_F()
148 *computation_layout.mutable_parameter_layout(0) = col_major; in TEST_F()
149 *computation_layout.mutable_parameter_layout(1) = row_major; in TEST_F()
338 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
377 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
415 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
466 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
529 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
531 *computation_layout.mutable_parameter_layout(1) = in TEST_F()
[all …]
H A Dhlo_module_util.cc70 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfig()
172 *module->mutable_entry_computation_layout()->mutable_parameter_layout(i) = in UpdateEntryComputationLayout()
H A Dcomputation_layout.h54 ShapeLayout* mutable_parameter_layout(int64_t param_no) { in mutable_parameter_layout() function
H A Dlayout_assignment.cc713 *body_layout.mutable_parameter_layout(0) = body_layout.result_layout(); in AddMandatoryConstraints()
724 *condition_layout.mutable_parameter_layout(0) = in AddMandatoryConstraints()
2120 callee_layout.mutable_parameter_layout(operand_no))) { in CalculateComputationLayout()
2368 ShapeLayout* param_layout = computation_layout->mutable_parameter_layout(i); in PropagateComputationLayouts()
H A Dhlo_module.cc575 entry_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfigFromShape()
H A Dhlo_computation.cc191 *config.mutable_entry_computation_layout()->mutable_parameter_layout( in ReplaceEntryComputationParameter()
H A Dhlo_parser.cc1002 ->mutable_parameter_layout(p) in ParseComputations()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment_test.cc79 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
120 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
164 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
166 *computation_layout.mutable_parameter_layout(1) = in TEST_F()
198 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
238 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
313 *computation_layout.mutable_parameter_layout(0) = in RunDotOutputFusion()
315 *computation_layout.mutable_parameter_layout(1) = in RunDotOutputFusion()
457 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
459 *computation_layout.mutable_parameter_layout(1) = in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_layout_assignment_test.cc75 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
77 *computation_layout.mutable_parameter_layout(1) = in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dhlo_test_base.h305 ->mutable_parameter_layout(param_no) in ForceParameterLayout()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_compile_op_support.cc86 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfig()