Searched refs:IsPortDimsN (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 516 bool IsPortDimsN(const NodeDef& node, int port, int n) const { in IsPortDimsN() function in tensorflow::grappler::__anonbd4a17fb0111::NodeProcessor
532 return IsPortDimsN(node, 0, n); in IsPortZeroDimsN()
541 return NodeProcessor::IsPortDimsN(node, port, 4) || in IsPortDimsFour()
1331 : IsPortDimsN(*input0, input0_port, n); in IsNDOperateWithMD()
1333 : IsPortDimsN(*input1, input1_port, m); in IsNDOperateWithMD()
1662 bool is_input0_scalar_vector_4d = IsPortDimsN(*input0, input0_port, 0) || in ShouldProcess()
1663 IsPortDimsN(*input0, input0_port, 1) || in ShouldProcess()
1664 IsPortDimsN(*input0, input0_port, 4); in ShouldProcess()
|