Home
last modified time | relevance | path

Searched refs:ConvertSoftmaxOp (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2697 class ConvertSoftmaxOp : public OpRewritePattern<OpTy> { class
6233 ConvertSoftmaxOp<TF::LogSoftmaxOp, true>, in PopulateLegalizeTfPatterns()
6234 ConvertSoftmaxOp<TF::SoftmaxOp, false>, ConvertSplitOp, ConvertSplitVOp, in PopulateLegalizeTfPatterns()