Searched defs:IsAlongAxis (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ | ||
| D | generic_layout_optimizer_transposer.cc | 1473 bool ReduceTransposer::IsAlongAxis(const Tensor& tensor, in IsAlongAxis() function in tensorflow::grappler::ReduceTransposer |
| D | layout_optimizer.cc | 1905 bool IsAlongAxis(const std::vector<int>& axis) const { in IsAlongAxis() function in tensorflow::grappler::__anon5a8ea6d90111::SqueezeProcessor |