Searched refs:mutable_parameter_layout (Results 1 – 9 of 9) 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()307 *computation_layout.mutable_parameter_layout(0) = in RunDotOutputFusion()309 *computation_layout.mutable_parameter_layout(1) = in RunDotOutputFusion()
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
572 *body_layout.mutable_parameter_layout(0) = body_layout.result_layout(); in AddMandatoryConstraints()579 *condition_layout.mutable_parameter_layout(0) = in AddMandatoryConstraints()1895 ShapeLayout* param_layout = computation_layout->mutable_parameter_layout(i); in PropagateComputationLayouts()
384 entry_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfigFromProto()
288 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfig()
536 ->mutable_parameter_layout(p) in ParseComputations()
70 *computation_layout.mutable_parameter_layout(0) = in TEST_F()72 *computation_layout.mutable_parameter_layout(1) = in TEST_F()158 *computation_layout.mutable_parameter_layout(0) = in TEST_F()227 *computation_layout.mutable_parameter_layout(0) = in TEST_F()309 *computation_layout.mutable_parameter_layout(constrained_param_no) = in TEST_F()
269 ->mutable_parameter_layout(param_no) in ForceParameterLayout()