experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #24651 from siju-samuel:patch-10

Browse Source 
PiperOrigin-RevId: 227579339
This commit is contained in:
TensorFlower Gardener 2019-01-02 14:17:50 -08:00
commit 869f7cc28f
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/kernels/softmax_op_functor.h
View File

@ -57,7 +57,6 @@ struct SoftmaxEigenImpl {
Eigen::DSizes<int, 2> one_by_class(1, num_classes); Eigen::DSizes<int, 2> one_by_class(1, num_classes);
#else #else
Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; Eigen::IndexList<Eigen::type2index<kClassDim> > along_class;
Eigen::IndexList<Eigen::type2index<1> > depth_dim;
Eigen::IndexList<int, Eigen::type2index<1> > batch_by_one; Eigen::IndexList<int, Eigen::type2index<1> > batch_by_one;
batch_by_one.set(0, batch_size); batch_by_one.set(0, batch_size);
Eigen::IndexList<Eigen::type2index<1>, int> one_by_class; Eigen::IndexList<Eigen::type2index<1>, int> one_by_class;