Searched refs:mutable_parameter_layout (Results 1 – 11 of 11) sorted by relevance
80 *computation_layout.mutable_parameter_layout(0) = in TEST_F()121 *computation_layout.mutable_parameter_layout(0) = in TEST_F()165 *computation_layout.mutable_parameter_layout(0) = in TEST_F()167 *computation_layout.mutable_parameter_layout(1) = in TEST_F()199 *computation_layout.mutable_parameter_layout(0) = in TEST_F()239 *computation_layout.mutable_parameter_layout(0) = in TEST_F()314 *computation_layout.mutable_parameter_layout(0) = in RunDotOutputFusion()316 *computation_layout.mutable_parameter_layout(1) = in RunDotOutputFusion()459 *computation_layout.mutable_parameter_layout(0) = in TEST_F()461 *computation_layout.mutable_parameter_layout(1) = in TEST_F()
112 *computation_layout.mutable_parameter_layout(0) = shape_layout; in TEST_F()113 *computation_layout.mutable_parameter_layout(1) = shape_layout; in TEST_F()147 *computation_layout.mutable_parameter_layout(0) = col_major; in TEST_F()148 *computation_layout.mutable_parameter_layout(1) = row_major; in TEST_F()372 *computation_layout.mutable_parameter_layout(0) = in TEST_F()413 *computation_layout.mutable_parameter_layout(0) = in TEST_F()451 *computation_layout.mutable_parameter_layout(0) = in TEST_F()502 *computation_layout.mutable_parameter_layout(0) = in TEST_F()566 *computation_layout.mutable_parameter_layout(0) = in TEST_F()568 *computation_layout.mutable_parameter_layout(1) = in TEST_F()[all …]
54 ShapeLayout* mutable_parameter_layout(int64 param_no) { in mutable_parameter_layout() function
70 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfig()
590 *body_layout.mutable_parameter_layout(0) = body_layout.result_layout(); in AddMandatoryConstraints()597 *condition_layout.mutable_parameter_layout(0) = in AddMandatoryConstraints()895 *FindOrDie(computation_layouts_, branch_comp).mutable_parameter_layout(0) = in CopyOperandIfLayoutsDiffer()2052 ShapeLayout* param_layout = computation_layout->mutable_parameter_layout(i); in PropagateComputationLayouts()
474 entry_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfigFromShape()
157 *config.mutable_entry_computation_layout()->mutable_parameter_layout( in ReplaceEntryComputationParameter()
859 ->mutable_parameter_layout(p) in ParseComputations()
72 *computation_layout.mutable_parameter_layout(0) = in TEST_F()74 *computation_layout.mutable_parameter_layout(1) = in TEST_F()160 *computation_layout.mutable_parameter_layout(0) = in TEST_F()229 *computation_layout.mutable_parameter_layout(0) = in TEST_F()311 *computation_layout.mutable_parameter_layout(constrained_param_no) = in TEST_F()
270 ->mutable_parameter_layout(param_no) in ForceParameterLayout()
84 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfig()