Searched refs:axis_to_reduce (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 112 Reduce::Reduce(const std::map<Axis, int>& axis_to_reduce, OperationType op_type, in Reduce() argument
119 auto it = axis_to_reduce.find(a); in Reduce()
120 if (it != axis_to_reduce.end()) { in Reduce()
162 const std::vector<Axis>& axis_to_reduce, in GetReduceKernelCode() argument
178 if (!HasAxis(axis_to_reduce, a)) { in GetReduceKernelCode()
183 const bool channels_reductin = HasAxis(axis_to_reduce, Axis::CHANNELS); in GetReduceKernelCode()
274 if (op_def.dst_tensors[0].HasAxis(a) && !HasAxis(axis_to_reduce, a)) { in GetReduceKernelCode()
305 for (int i = 0; i < axis_to_reduce.size(); ++i) { in GetReduceKernelCode()
306 const auto& axis = axis_to_reduce[i]; in GetReduceKernelCode()
307 const int index = axis_to_reduce.size() - 1 - i; in GetReduceKernelCode()
[all …]
|
| D | reduce.h | 33 Reduce(const std::map<Axis, int>& axis_to_reduce, OperationType op_type,
52 const std::vector<Axis>& axis_to_reduce,
58 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, const BHWC& src_shape,
62 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | simple_selectors.cc | 161 std::unique_ptr<GPUOperation> SelectReduce(const std::set<Axis>& axis_to_reduce, in SelectReduce() argument
167 CreateReduce(axis_to_reduce, src_shape, op_type, op_def, gpu_info)); in SelectReduce()
|
| D | simple_selectors.h | 72 std::unique_ptr<GPUOperation> SelectReduce(const std::set<Axis>& axis_to_reduce,
|