Searched refs:kSparseSoftmaxCrossEntropyWithLogitsInputTensorNum (Results 1 – 2 of 2) sorted by relevance
116 CheckCNodeInputSize(sparse_softmax_node, kSparseSoftmaxCrossEntropyWithLogitsInputTensorNum); in CreateSoftmaxCrossEntropyWithLogits()164 CheckCNodeInputSize(sparse_softmax_node, kSparseSoftmaxCrossEntropyWithLogitsInputTensorNum); in CreateReduceMean()256 CheckCNodeInputSize(sparse_softmax_node, kSparseSoftmaxCrossEntropyWithLogitsInputTensorNum); in CreateTile()304 CheckCNodeInputSize(sparse_softmax_node, kSparseSoftmaxCrossEntropyWithLogitsInputTensorNum); in CreateRealDiv()403 CheckCNodeInputSize(sparse_softmax_node, kSparseSoftmaxCrossEntropyWithLogitsInputTensorNum); in Process()444 CheckCNodeInputSize(sparse_softmax_node_grad, kSparseSoftmaxCrossEntropyWithLogitsInputTensorNum); in Process()521 CheckCNodeInputSize(sparse_softmax_node, kSparseSoftmaxCrossEntropyWithLogitsInputTensorNum); in Process()561 CheckCNodeInputSize(sparse_softmax_node_grad, kSparseSoftmaxCrossEntropyWithLogitsInputTensorNum); in Process()
108 constexpr size_t kSparseSoftmaxCrossEntropyWithLogitsInputTensorNum = 2; variable