Searched refs:IsAlongAxis (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 1846 bool IsAlongAxis(const std::vector<int>& axis) const { in IsAlongAxis() function in tensorflow::grappler::__anonbd4a17fb0111::SqueezeProcessor
1861 bool IsAlongHW() const { return IsAlongAxis({1, 2}); } in IsAlongHW()
1862 bool IsAlongNHW() const { return IsAlongAxis({0, 1, 2}); } in IsAlongNHW()
1903 bool IsAlongAxis(const std::vector<int>& axis) const { in IsAlongAxis() function in tensorflow::grappler::__anonbd4a17fb0111::ReduceProcessor
1925 bool IsAlongAllFourDims() const { return IsAlongAxis({0, 1, 2, 3}); } in IsAlongAllFourDims()
1927 bool IsAlongHWC() const { return IsAlongAxis({1, 2, 3}); } in IsAlongHWC()
1929 bool IsAlongNHW() const { return IsAlongAxis({0, 1, 2}); } in IsAlongNHW()
1931 bool IsAlongHW() const { return IsAlongAxis({1, 2}); } in IsAlongHW()
1933 bool IsAlongC() const { return IsAlongAxis({3}); } in IsAlongC()
|