Home
last modified time | relevance | path

Searched refs:SparseSoftmaxXentWithLogitsOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_xent_op.cc51 class SparseSoftmaxXentWithLogitsOp : public OpKernel { class
53 explicit SparseSoftmaxXentWithLogitsOp(OpKernelConstruction* context) in SparseSoftmaxXentWithLogitsOp() function in tensorflow::SparseSoftmaxXentWithLogitsOp
120 SparseSoftmaxXentWithLogitsOp<Dev##Device, T, Index>);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsoftmax_op.cc175 class SparseSoftmaxXentWithLogitsOp : public XlaOpKernel { class
177 explicit SparseSoftmaxXentWithLogitsOp(OpKernelConstruction* ctx) in SparseSoftmaxXentWithLogitsOp() function in tensorflow::__anon9b27491f0111::SparseSoftmaxXentWithLogitsOp
241 SparseSoftmaxXentWithLogitsOp);