Searched refs:batch_only (Results 1 – 2 of 2) sorted by relevance
77 Eigen::array<int, 1> batch_only; in Compute() local78 batch_only[0] = batch_size; in Compute()89 Eigen::IndexList<int> batch_only; in Compute() local90 batch_only.set(0, batch_size); in Compute()96 scratch.reshape(batch_only).device(d) = in Compute()104 scratch.reshape(batch_only).device(d) = backprop.exp().sum(along_class); in Compute()
194 Eigen::array<int, 1> batch_only; in Compute() local195 batch_only[0] = batch_size; in Compute()206 Eigen::IndexList<int> batch_only; in Compute() local207 batch_only.set(0, batch_size); in Compute()