Searched defs:reduction_dims (Results 1 – 6 of 6) sorted by relevance
80 const int32_t reduction_dims = reduction_indices_flat.size(); in GetReducedIndices() local132 const int32_t reduction_dims = reduction_indices_flat.size(); in Compute() local
102 reduction_dims; variable204 Eigen::IndexList<Eigen::type2index<1> > reduction_dims; in CuboidMaxPooling() local
427 std::vector<Tmultiples> reduction_dims; in HandleCaseImpl() local
304 std::vector<int64_t> reduction_dims(input_dims - 1); in Compile() local
757 NodeDef reduction_dims; in QuantizeNodes() local
1710 def reduction_dims(shape: ShapeType, dims: Optional[Sequence]) -> Tuple[int, ...]: function