Home
last modified time | relevance | path

Searched defs:ReduceMax (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dreduce_max.h35 ReduceMax() : Reduce(kNameReduceMax) { InitIOName({"input_x", "axis"}, {"y"}); } in ReduceMax() function
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_reduce_max.py24 class ReduceMax(nn.Cell): class
/third_party/mindspore/tests/st/ops/gpu/
Dtest_reduce_max_op.py67 class ReduceMax(nn.Cell): class
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dreduce_fp32.c182 int ReduceMax(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data… in ReduceMax() function
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_op_converter.cc335 MS_TRT_CONVERTER_FUNC_REG(ReduceMax) { return AddReduceLayer(node, context, nvinfer1::ReduceOperati… in MS_TRT_CONVERTER_FUNC_REG() argument
/third_party/mindspore/mindspore/ops/operations/
Dmath_ops.py742 class ReduceMax(_Reduce): class