Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1242 auto tf_logsoftmax_op = cast<TF::LogSoftmaxOp>(op); in matchAndRewrite()
Dlegalize_tfl.cc1918 auto tfl_logsoftmax_op = cast<TFL::LogSoftmaxOp>(op); in matchAndRewrite()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_test.cc1122 TEST_F(AutoMixedPrecisionTest, LogSoftmaxOp) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc6233 ConvertSoftmaxOp<TF::LogSoftmaxOp, true>, in PopulateLegalizeTfPatterns()