Searched defs:SoftmaxOp (Results 1 – 3 of 3) sorted by relevance
37 struct SoftmaxOp { struct38 SoftmaxOp(const float beta) : beta_(beta) {} in SoftmaxOp() function41 typename Input::Dimensions dimensions(const Input& input) const { in dimensions()46 void eval(const Input& input, Output& output, const Device& device) const { in eval()
56 class SoftmaxOp : public OpKernel { class58 explicit SoftmaxOp(OpKernelConstruction* context) : OpKernel(context) { in SoftmaxOp() function in tensorflow::SoftmaxOp
33 class SoftmaxOp : public XlaOpKernel { class35 explicit SoftmaxOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SoftmaxOp() function in tensorflow::__anon9b27491f0111::SoftmaxOp