Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsoftmax_op.cc49 class SoftmaxOp : public OpKernel { class
51 explicit SoftmaxOp(OpKernelConstruction* context) : OpKernel(context) { in SoftmaxOp() function in tensorflow::SoftmaxOp
77 SoftmaxOp<CPUDevice, T>);
84 SoftmaxOp<CPUDevice, T>);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsoftmax_op.cc49 class SoftmaxOp : public XlaOpKernel { class
51 explicit SoftmaxOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SoftmaxOp() function in tensorflow::__anon0443f5bd0111::SoftmaxOp
113 REGISTER_XLA_OP(Name("Softmax"), SoftmaxOp);
114 REGISTER_XLA_OP(Name("LogSoftmax"), SoftmaxOp);
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1228 auto tf_softmax_op = cast<TF::SoftmaxOp>(op); in matchAndRewrite()
Dlegalize_tfl.cc1904 auto tfl_softmax_op = cast<TFL::SoftmaxOp>(op); in matchAndRewrite()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_test.cc1146 TEST_F(AutoMixedPrecisionTest, SoftmaxOp) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1154 static LogicalResult Verify(SoftmaxOp op) { in Verify()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc6234 ConvertSoftmaxOp<TF::SoftmaxOp, false>, ConvertSplitOp, ConvertSplitVOp, in PopulateLegalizeTfPatterns()