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

  /external/tensorflow/tensorflow/core/kernels/
xent_op.h 68 Eigen::array<int, 1> batch_only; local
69 batch_only[0] = batch_size;
80 Eigen::IndexList<int> batch_only; local
81 batch_only.set(0, batch_size);
87 scratch.reshape(batch_only).device(d) = logits.maximum(along_class);
93 scratch.reshape(batch_only).device(d) = backprop.exp().sum(along_class);
sparse_xent_op.h 172 Eigen::array<int, 1> batch_only; local
173 batch_only[0] = batch_size;
184 Eigen::IndexList<int> batch_only; local
185 batch_only.set(0, batch_size);

Completed in 135 milliseconds