Searched refs:SoftmaxOp (Results 1 – 7 of 7) sorted by relevance
49 class SoftmaxOp : public OpKernel { class51 explicit SoftmaxOp(OpKernelConstruction* context) : OpKernel(context) { in SoftmaxOp() function in tensorflow::SoftmaxOp77 SoftmaxOp<CPUDevice, T>);84 SoftmaxOp<CPUDevice, T>);
49 class SoftmaxOp : public XlaOpKernel { class51 explicit SoftmaxOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SoftmaxOp() function in tensorflow::__anon0443f5bd0111::SoftmaxOp113 REGISTER_XLA_OP(Name("Softmax"), SoftmaxOp);114 REGISTER_XLA_OP(Name("LogSoftmax"), SoftmaxOp);
1228 auto tf_softmax_op = cast<TF::SoftmaxOp>(op); in matchAndRewrite()
1904 auto tfl_softmax_op = cast<TFL::SoftmaxOp>(op); in matchAndRewrite()
1146 TEST_F(AutoMixedPrecisionTest, SoftmaxOp) { in TEST_F() argument
1154 static LogicalResult Verify(SoftmaxOp op) { in Verify()
6234 ConvertSoftmaxOp<TF::SoftmaxOp, false>, ConvertSplitOp, ConvertSplitVOp, in PopulateLegalizeTfPatterns()