Searched refs:GetSizesFromShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 92 std::map<Axis, int> GetSizesFromShape(const std::set<Axis>& axis, in GetSizesFromShape() function
101 std::map<Axis, int> GetSizesFromShape(const std::set<Axis>& axis, in GetSizesFromShape() function
478 return Reduce(GetSizesFromShape(axis_to_reduce, src_shape), op_type, in CreateReduce()
485 return Reduce(GetSizesFromShape(axis_to_reduce, src_shape), op_type, in CreateReduce()
|