Searched defs:ReduceProd (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | reduce_impl.cuh | 25 ReduceProd = 1, enumerator
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_reduce_prod_op.py | 25 class ReduceProd(nn.Cell): class
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | reduce_cpu_kernel.cc | 146 void ReduceProd(const T *in, T *out, size_t start, size_t end, TransposeIterator *iter) { in ReduceProd() function
175 void ReduceProd(const Eigen::half *in, Eigen::half *out, size_t start, size_t end, TransposeIterato… in ReduceProd() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/ |
| D | gen_lite_ops.h | 1973 ReduceProd() : BaseOperator(kNameReduceProd) {} in ReduceProd() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/ |
| D | trt_op_converter.cc | 348 MS_TRT_CONVERTER_FUNC_REG(ReduceProd) { return AddReduceLayer(node, context, nvinfer1::ReduceOperat… in MS_TRT_CONVERTER_FUNC_REG() argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/ |
| D | cust_aicpu_kernel.ini | 2779 [ReduceProd] section
2780 opInfo.subTypeOfInferShape=1
2781 opInfo.opsFlag=OPS_FLAG_CLOSE
2782 opInfo.engine=DNN_VM_AICPU
2783 opInfo.flagPartial=False
2784 opInfo.computeCost=100
2785 opInfo.flagAsync=False
2786 opInfo.opKernelLib=CUSTAICPUKernel
2787 opInfo.formatAgnostic=False
2788 opInfo.kernelSo=libcust_cpu_kernels.so
[all …]
|