Searched defs:reduction_dims (Results 1 – 6 of 6) 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
427 std::vector<Tmultiples> reduction_dims; in HandleCaseImpl() local
300 std::vector<int64> reduction_dims(input_dims - 1); in Compile() local
130 ArrayRef<int64_t> reduction_dims) { in GetTransposeMapForReduction()940 SmallVector<int, 4> reduction_dims; in matchAndRewrite() local1273 ArrayRef<int64_t> reduction_dims) { in GetReduceOpInitTensorDynSizes()1333 SmallVector<int64_t, 4> reduction_dims = Extract1DVector(op.dimensions()); in matchAndRewrite() local
757 NodeDef reduction_dims; in QuantizeNodes() local