Searched defs:axis_to_reduce (Results 1 – 2 of 2) sorted by relevance
112 Reduce::Reduce(const std::map<Axis, int>& axis_to_reduce, OperationType op_type, in Reduce()162 const std::vector<Axis>& axis_to_reduce, in GetReduceKernelCode()475 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, const BHWC& src_shape, in CreateReduce()482 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, in CreateReduce()
161 std::unique_ptr<GPUOperation> SelectReduce(const std::set<Axis>& axis_to_reduce, in SelectReduce()