Searched defs:sparse_softmax_node (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 102 void GetDepthAndBatchSizeFromSparseSoftmaxNode(const AnfNodePtr &sparse_softmax_node, int64_t *batc… in GetDepthAndBatchSizeFromSparseSoftmaxNode()
116 CNodePtr CreateOneHot(const FuncGraphPtr &graph, const CNodePtr &sparse_softmax_node, const Pattern… in CreateOneHot()
169 …CreateSoftmaxCrossEntropyWithLogits(const FuncGraphPtr &graph, const CNodePtr &sparse_softmax_node, in CreateSoftmaxCrossEntropyWithLogits()
235 CNodePtr CreateReduceMean(const FuncGraphPtr &graph, const CNodePtr &sparse_softmax_node, in CreateReduceMean()
324 CNodePtr CreateTile(const FuncGraphPtr &graph, const CNodePtr &sparse_softmax_node, const CNodePtr … in CreateTile()
386 CNodePtr CreateRealDiv(const FuncGraphPtr &graph, const CNodePtr &sparse_softmax_node, const AnfNod… in CreateRealDiv()
432 CNodePtr CreateMul(const FuncGraphPtr &graph, const CNodePtr &sparse_softmax_node, in CreateMul()
504 …ulInput(const FuncGraphPtr &graph, const CNodePtr &mul_node, const AnfNodePtr &sparse_softmax_node, in CreateMulInput()
554 auto sparse_softmax_node = node->cast<CNodePtr>(); in Process() local
603 auto sparse_softmax_node = GetSparseNode(depend_node, kIndex2); in Process() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/ |
| D | ir_pass.cc | 168 CNodePtr Run(const CNodePtr &mul_node, const AnfNodePtr &sparse_softmax_node) { in Run()
213 void GetDepthAndBatchSizeFromSparseSoftmaxNode(const AnfNodePtr &sparse_softmax_node) { in GetDepthAndBatchSizeFromSparseSoftmaxNode()
227 CNodePtr CreateOneHot(const CNodePtr &sparse_softmax_node) { in CreateOneHot()
260 …CNodePtr CreateSoftmaxCrossEntropyWithLogits(const CNodePtr &sparse_softmax_node, const CNodePtr &… in CreateSoftmaxCrossEntropyWithLogits()
299 CNodePtr CreateTile(const CNodePtr &sparse_softmax_node, const CNodePtr &mul_node) { in CreateTile()
345 CNodePtr CreateRealDiv(const CNodePtr &sparse_softmax_node, const AnfNodePtr &tile_node) { in CreateRealDiv()
392 CNodePtr CreateMulInput(const CNodePtr &mul_node, const AnfNodePtr &sparse_softmax_node, in CreateMulInput()
440 auto sparse_softmax_node = mul_node->input(kIndex1); in GradSparseSoftmaxCrossEntropyWithLogitsUnifyMindIR() local
|