Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsparse_softmax_cross_entropy_with_logits.h34 SparseSoftmaxCrossEntropyWithLogits() : BaseOperator(kNameSparseSoftmaxCrossEntropyWithLogits) {} in SparseSoftmaxCrossEntropyWithLogits() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dnn_norm_ops_proto.cc237 IMPLEMT_VERIFIER(SparseSoftmaxCrossEntropyWithLogits, SparseSoftmaxCrossEntropyWithLogitsInfer) { in IMPLEMT_VERIFIER() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/
Dfunc_builder.cc207 NodePtr FuncBuilder::SparseSoftmaxCrossEntropyWithLogits(const NodePtrList &inputs, const expander:… in SparseSoftmaxCrossEntropyWithLogits() function in mindspore::pynative::autograd::FuncBuilder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/expander/core/
Demitter.cc580 NodePtr Emitter::SparseSoftmaxCrossEntropyWithLogits(const NodePtrList &inputs, const DAttr &attrs,… in SparseSoftmaxCrossEntropyWithLogits() function in mindspore::expander::Emitter
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/
Dcust_aicpu_kernel.ini6157 [SparseSoftmaxCrossEntropyWithLogits] section
6158 opInfo.subTypeOfInferShape=1
6159 opInfo.opsFlag=OPS_FLAG_CLOSE
6160 opInfo.engine=DNN_VM_AICPU
6161 opInfo.flagPartial=False
6162 opInfo.computeCost=100
6163 opInfo.flagAsync=False
6164 opInfo.opKernelLib=CUSTAICPUKernel
6165 opInfo.formatAgnostic=False
6166 opInfo.userDefined=True
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dnn_ops.py2161 class SparseSoftmaxCrossEntropyWithLogits(Primitive): class