Searched defs:axis_to_reduce (Results 1 – 2 of 2) sorted by relevance
128 Reduce::Reduce(const std::map<Axis, int>& axis_to_reduce, OperationType op_type, in Reduce()179 const std::vector<Axis>& axis_to_reduce, in GetReduceKernelCode()525 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, const BHWC& src_shape, in CreateReduce()532 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, in CreateReduce()
193 std::unique_ptr<GPUOperation> SelectReduce(const std::set<Axis>& axis_to_reduce, in SelectReduce()