Home
last modified time | relevance | path

Searched refs:MustPreserve (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc578 bool MustPreserve() const { in MustPreserve() function in tensorflow::grappler::__anon9cd52bfd0111::NodeProcessor
600 return !MustPreserve() && IsNHWC() && IsPortZeroDimsFour(*node_) && in ShouldProcess()
1075 if (MustPreserve()) { in ShouldProcess()
1102 return !MustPreserve() && IsNHWC() && IsPortZeroDimsFour(*node_) && in ShouldProcess()
1264 return !MustPreserve() && IsNHWC() && IsPortDimsFour(*data_input, port) && in ShouldProcess()
1284 return !MustPreserve() && IsPortZeroDimsFour(*node_) && HasOutputs() && in ShouldProcess()
1346 return !MustPreserve() && IsPortZeroDimsFour(*node_) && HasOutputs() && in ShouldProcess()
1525 return !MustPreserve() && HasOutputs() && IsNodeAfterNCHWToNHWC() && in ShouldProcess()
1558 return !MustPreserve() && HasOutputs() && IsNodeAfterNCHWToNHWC() && in ShouldProcess()
1596 return !MustPreserve() && IsPortZeroDimsFour(*node_) && HasOutputs() && in ShouldProcess()
[all …]
Dauto_mixed_precision.cc958 bool MustPreserve(const NodeDef& node) const;
1078 << (MustPreserve(node) in LogSkippedNode()
1083 bool AutoMixedPrecisionImpl::MustPreserve(const NodeDef& node) const { in MustPreserve() function in tensorflow::grappler::__anon09ee8e770111::AutoMixedPrecisionImpl
1212 if (!MustPreserve(node) && IsOnGPU(node) && in Optimize()