Searched refs:IsPortDimsFour (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 540 bool IsPortDimsFour(const NodeDef& node, int port) const { in IsPortDimsFour() function in tensorflow::grappler::__anonbd4a17fb0111::NodeProcessor
1035 if (IsNHWC() && IsPortDimsFour(*input, port)) { in ShouldProcess()
1214 return !MustPreserve() && IsNHWC() && IsPortDimsFour(*data_input, port) && in ShouldProcess()
1312 if (IsPortDimsFour(*input0, input0_port)) { in GetInputPos()
1315 if (IsPortDimsFour(*input1, input1_port)) { in GetInputPos()
1330 bool input0_is_n = (n == 4) ? IsPortDimsFour(*input0, input0_port) in IsNDOperateWithMD()
1332 bool input1_is_m = (m == 4) ? IsPortDimsFour(*input1, input1_port) in IsNDOperateWithMD()
1476 IsPortDimsFour(*input1, port) && IsOnGPU(); in ShouldProcess()
1497 if (IsPortDimsFour(*input, port) && in IdentityNProcessor()
1569 if (IsPortDimsFour(*input_node, port) && in IsEveryInputAfterNCHWToNHWC()
[all …]
|