Home
last modified time | relevance | path

Searched defs:ArgMaxWithValue (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/pynative/pyboost/
Dtest_simply_infer.py17 from mindspore.ops.function.array_func import arg_max_with_value_ as ArgMaxWithValue unknown
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dtest_argmaxwithvalue.py25 class ArgMaxWithValue(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/
Dtest_argmax_with_value.py24 class ArgMaxWithValue(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/pynative/
Dtest_pynative_layernorm_input_and_argmaxwithvalue.py147 class ArgMaxWithValue(Cell): class
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.h179 ArgMaxWithValue() : BaseOperator(kNameArgMaxWithValue) {} in ArgMaxWithValue() function
/third_party/mindspore/mindspore-src/source/config/
Dsuper_bar_config.json197 "ArgMaxWithValue": { object
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/
Dcust_aicpu_kernel.ini4332 [ArgMaxWithValue] section
4333 opInfo.subTypeOfInferShape=1
4334 opInfo.opsFlag=OPS_FLAG_CLOSE
4335 opInfo.engine=DNN_VM_AICPU
4336 opInfo.flagPartial=False
4337 opInfo.computeCost=100
4338 opInfo.flagAsync=False
4339 opInfo.opKernelLib=CUSTAICPUKernel
4340 opInfo.formatAgnostic=False
4341 opInfo.userDefined=True
[all …]