Searched defs:ReduceSum (Results 1 – 6 of 6) sorted by relevance
35 ReduceSum() : Reduce(kNameReduceSum) { InitIOName({"x", "axis"}, {"y"}); } in ReduceSum() function
20 ReduceSum = P.ReduceSum(keep_dims=True) variable
90 class ReduceSum(nn.Cell): class
100 int ReduceSum(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data… in ReduceSum() function
333 MS_TRT_CONVERTER_FUNC_REG(ReduceSum) { return AddReduceLayer(node, context, nvinfer1::ReduceOperati… in MS_TRT_CONVERTER_FUNC_REG() argument
542 class ReduceSum(_Reduce): class