Searched defs:reshape_dim (Results 1 – 3 of 3) sorted by relevance
493 Eigen::DSizes<Eigen::DenseIndex, NDIM> reshape_dim; in HandleReduce() local
268 const int64_t reshape_dim = MathUtil::IPow(dim, count); in StrideOrInflate() local
4514 auto reshape_dim = DimsAdapter(input_dim.nbDims, in ConvertFullyConnectedImpl() local