Searched refs:UpdateOrTransformParamInput (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 720 Status UpdateOrTransformParamInput(int param_index, const string& op, in UpdateOrTransformParamInput() function in tensorflow::grappler::__anonbd4a17fb0111::NodeProcessor
1014 return UpdateOrTransformParamInput(0, "DataFormatVecPermute", DT_INT32); in CustomizedProcessing()
1150 return UpdateOrTransformParamInput(0, "DataFormatVecPermute", DT_INT32); in CustomizedProcessing()
1195 UpdateOrTransformParamInput(i, "DataFormatVecPermute", DT_INT32)); in CustomizedProcessing()
1221 UpdateOrTransformParamInput(i, "DataFormatVecPermute", DT_INT32)); in CustomizedProcessing()
1444 return UpdateOrTransformParamInput(axis_node_pos_, "DataFormatDimMap", in CustomizedProcessing()
1461 return UpdateOrTransformParamInput(0, "DataFormatVecPermute", dtype); in CustomizedProcessing()
1588 return UpdateOrTransformParamInput(1, "DataFormatVecPermute", dtype); in CustomizedProcessing()
1600 return UpdateOrTransformParamInput(1, "DataFormatDimMap", dtype); in CustomizedProcessing()
1625 return UpdateOrTransformParamInput(axis_node_pos_, "DataFormatDimMap", in CustomizedProcessing()
[all …]
|