Searched refs:CrossEntropyWithLogits (Results 1 – 1 of 1) sorted by relevance
91 std::pair<xla::XlaOp, xla::XlaOp> CrossEntropyWithLogits( in CrossEntropyWithLogits() function167 CrossEntropyWithLogits(ctx, type, xla_type, logits, labels); in Compile()233 std::tie(loss, backprop) = CrossEntropyWithLogits( in Compile()