Searched refs:IsNodeAfterNCHWToNHWC (Results 1 – 1 of 1) sorted by relevance
1235 IsNodeAfterNCHWToNHWC() && IsOnGPU(); in ShouldProcess()1238 bool IsNodeAfterNCHWToNHWC(const NodeDef& node) const { in IsNodeAfterNCHWToNHWC() function in tensorflow::grappler::__anonbd4a17fb0111::AgnosticNodeProcessor1275 bool IsNodeAfterNCHWToNHWC() const { return IsNodeAfterNCHWToNHWC(*node_); } in IsNodeAfterNCHWToNHWC() function in tensorflow::grappler::__anonbd4a17fb0111::AgnosticNodeProcessor1297 IsNodeAfterNCHWToNHWC() && in ShouldProcess()1475 return !MustPreserve() && HasOutputs() && IsNodeAfterNCHWToNHWC() && in ShouldProcess()1498 ((IsNodeAfterNCHWToNHWC(*input) && is_agnostic) || in IdentityNProcessor()1508 return !MustPreserve() && HasOutputs() && IsNodeAfterNCHWToNHWC() && in ShouldProcess()1570 ((IsNodeAfterNCHWToNHWC(*input_node) && is_agnostic) || in IsEveryInputAfterNCHWToNHWC()1806 return !MustPreserve() && HasOutputs() && IsNodeAfterNCHWToNHWC() && in ShouldProcess()1875 return !MustPreserve() && HasOutputs() && IsNodeAfterNCHWToNHWC() && in ShouldProcess()