Home
last modified time | relevance | path

Searched defs:SoftmaxXentWithLogitsOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dxent_op.cc38 class SoftmaxXentWithLogitsOp : public OpKernel { class
40 explicit SoftmaxXentWithLogitsOp(OpKernelConstruction* context) in SoftmaxXentWithLogitsOp() function in tensorflow::SoftmaxXentWithLogitsOp
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsoftmax_op.cc142 class SoftmaxXentWithLogitsOp : public XlaOpKernel { class
144 explicit SoftmaxXentWithLogitsOp(OpKernelConstruction* ctx) in SoftmaxXentWithLogitsOp() function in tensorflow::__anon9b27491f0111::SoftmaxXentWithLogitsOp