Searched refs:SoftmaxOp (Results 1 – 3 of 3) sorted by relevance
56 class SoftmaxOp : public OpKernel { class58 explicit SoftmaxOp(OpKernelConstruction* context) : OpKernel(context) { in SoftmaxOp() function in tensorflow::SoftmaxOp84 SoftmaxOp<CPUDevice, T>);93 SoftmaxOp<CPUDevice, T>);101 SoftmaxOp<SYCLDevice, float>);104 SoftmaxOp<SYCLDevice, double>);
37 struct SoftmaxOp { struct38 SoftmaxOp(const float beta) : beta_(beta) {} in SoftmaxOp() function85 EIGEN_ALWAYS_INLINE static const TensorCustomUnaryOp<const SoftmaxOp,93 const SoftmaxOp op(beta); in SoftMax()
33 class SoftmaxOp : public XlaOpKernel { class35 explicit SoftmaxOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SoftmaxOp() function in tensorflow::__anon9b27491f0111::SoftmaxOp88 REGISTER_XLA_OP(Name("Softmax"), SoftmaxOp);89 REGISTER_XLA_OP(Name("LogSoftmax"), SoftmaxOp);