Searched refs:AddNodePermConst (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 408 NodeDef* AddNodePermConst(const string& name, const string& device, in AddNodePermConst() function in tensorflow::grappler::__anonbd4a17fb0111::GraphProcessor
885 auto const_node = AddNodePermConst(name, device, {0, 3, 1, 2}); in AddNodePermNHWCToNCHW()
895 auto const_node = AddNodePermConst( in AddNodePermNCHWToNHWC()
1982 return AddNodePermConst(LayoutOptimizerNode(kPermNHWCToNCHW), "", in AddNodePermNHWCToNCHW()
1987 return AddNodePermConst(LayoutOptimizerNode(kPermNCHWToNHWC), "", in AddNodePermNCHWToNHWC()
|