Home
last modified time | relevance | path

Searched defs:ReduceProd (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dreduce_prod.h35 ReduceProd() : Reduce(kNameReduceProd) { InitIOName({"input_x", "axis"}, {"y"}); } in ReduceProd() function
/third_party/mindspore/tests/st/ops/gpu/
Dtest_reduce_prod_op.py65 class ReduceProd(nn.Cell): class
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dreduce_fp32.c307 int ReduceProd(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_dat… in ReduceProd() function
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_op_converter.cc337 MS_TRT_CONVERTER_FUNC_REG(ReduceProd) { return AddReduceLayer(node, context, nvinfer1::ReduceOperat… in MS_TRT_CONVERTER_FUNC_REG() argument
/third_party/mindspore/mindspore/ops/operations/
Dmath_ops.py907 class ReduceProd(_Reduce): class