HomeSort by relevance Sort by last modified time
    Searched refs:loss_out (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
sparse_xent_op.cc 80 Tensor* loss_out = nullptr; variable
82 {1}, 0, labels.shape(), &loss_out));
94 labels.vec<Index>(), scratch.vec<T>(), loss_out->vec<T>(),
xent_op.cc 62 Tensor* loss_out = nullptr; variable
65 0, TensorShape({logits_in.dim_size(0)}), &loss_out));
73 labels_in.matrix<T>(), scratch.matrix<T>(), loss_out->vec<T>(),

Completed in 80 milliseconds