Home
last modified time | relevance | path

Searched defs:SoftmaxOp (Results 1 – 3 of 3) 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
/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
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_test.cc1146 TEST_F(AutoMixedPrecisionTest, SoftmaxOp) { in TEST_F() argument