Searched refs:kSoftmaxCrossEntropyWithLogitsOpName (Results 1 – 2 of 2) sorted by relevance
311 constexpr auto kSoftmaxCrossEntropyWithLogitsOpName = "SoftmaxCrossEntropyWithLogits"; variable
117 …NodePtr> inputs = {NewValueNode(std::make_shared<Primitive>(kSoftmaxCrossEntropyWithLogitsOpName)), in CreateSoftmaxCrossEntropyWithLogits()