Home
last modified time | relevance | path

Searched refs:kPrimSoftmaxCrossEntropyWithLogits (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dnn_norm_ops_declare.cc39 REG_ADPT_DESC(SoftmaxCrossEntropyWithLogits, prim::kPrimSoftmaxCrossEntropyWithLogits->name(),
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_expander.cc90 {kGPUDevice, OpLevel_1, prim::kPrimSoftmaxCrossEntropyWithLogits}, in GetExpandOps()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc36 …rimDepend, prim::kPrimTupleGetItem, prim::kPrimAdd, prim::kPrimSoftmaxCrossEntropyWithLogits,
/third_party/mindspore/mindspore/core/base/
Dcore_ops.h336 inline const PrimitivePtr kPrimSoftmaxCrossEntropyWithLogits = variable