Searched defs:log_softmax (Results 1 – 3 of 3) sorted by relevance
159 auto log_softmax = builder.create<mlir::TF::SubOp>( in ComputeLogSoftmax() local169 bool log_softmax) { in ComputeShardedSoftmax()579 const mlir::Value log_softmax = in ExpandOp() local
515 log_softmax = nn.log_softmax variable
3918 def log_softmax(logits, axis=None, name=None, dim=None): function