Searched defs:reduction_dims (Results 1 – 5 of 5) sorted by relevance
112 Eigen::array<int, 2> reduction_dims; variable127 Eigen::type2index<3> > >::type reduction_dims; variable242 Eigen::array<int, 1> reduction_dims; in CuboidMaxPooling() local247 Eigen::IndexList<Eigen::type2index<1> > reduction_dims; in CuboidMaxPooling() local
79 const int32 reduction_dims = reduction_indices_flat.size(); in GetReducedIndices() local131 const int32 reduction_dims = reduction_indices_flat.size(); in Compute() local
360 std::vector<Tmultiples> reduction_dims; in HandleCaseImpl() local
174 std::vector<int64> reduction_dims(input_dims - 1); in Compile() local
752 NodeDef reduction_dims; in QuantizeNodes() local