Searched defs:reduce_dim (Results 1 – 5 of 5) sorted by relevance
47 std::vector<int64_t> reduce_dim() const { return reduce_dim_; } in reduce_dim() function
45 std::vector<int64_t> ReduceMethod::reduce_dim() { in reduce_dim() function in mindspore::parallel::ReduceMethod319 std::vector<int64_t> ArgMaxWithValueInfo::reduce_dim() { in reduce_dim() function in mindspore::parallel::ArgMaxWithValueInfo425 std::vector<int64_t> ArgmaxInfo::reduce_dim() { in reduce_dim() function in mindspore::parallel::ArgmaxInfo501 std::vector<int64_t> SquareSumAllInfo::reduce_dim() { in reduce_dim() function in mindspore::parallel::SquareSumAllInfo
83 std::vector<int64_t> ReduceBaseMethod::reduce_dim() { in reduce_dim() function in mindspore::parallel::ReduceBaseMethod263 std::vector<int64_t> SumExtInfo::reduce_dim() { in reduce_dim() function in mindspore::parallel::SumExtInfo
136 const size_t shape_size, const size_t reduce_dim, const size_t dim_val) { in ReduceSum()170 const size_t shape_size, const size_t reduce_dim, const size_t dim_val) { in ReduceSum()204 const size_t shape_size, const size_t reduce_dim, const size_t dim_val, in CalReduceSum()
307 int reduce_dim = nframe; in MultiMarginLoss() local333 int reduce_dim = nframe; in MultiMarginLoss() local