Searched defs:softmax_node (Results 1 – 4 of 4) sorted by relevance
187 auto softmax_node = pass.NewCNode(inputs, graph); in CreateSoftmaxCrossEntropyWithLogits() local520 CNodePtr softmax_node; in CreateMulInput() local566 CNodePtr softmax_node; in Process() local611 CNodePtr softmax_node; in Process() local679 CNodePtr softmax_node; in Process() local716 CNodePtr softmax_node; in Process() local
271 auto softmax_node = ir_bprop_->ad_param()->tape_->FuncGraph::NewCNode(inputs); in CreateSoftmaxCrossEntropyWithLogits() local399 auto softmax_node = CreateSoftmaxCrossEntropyWithLogits(sparse_softmax_cnode, one_hot_node); in CreateMulInput() local
214 auto softmax_node = CreateSoftmaxCrossEntropyWithLogits(inputs, one_hot_node); in CreateMulInput() local
851 auto softmax_node = GetSoftmax(softmax_cnode); in PreposeWithSoftmax() local