Searched refs:reshape_dim (Results 1 – 4 of 4) sorted by relevance
83 reshape_dim = [100, 100]86 num_elements = num_epochs * reshape_dim[0]92 input_data.reshape(*reshape_dim)).repeat(num_epochs))
63 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& reshape_dim) const {64 out.device(d) = in.sum(reduce_dim).reshape(reshape_dim);
76 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& reshape_dim) const;493 Eigen::DSizes<Eigen::DenseIndex, NDIM> reshape_dim; in HandleReduce() local500 reshape_dim[i] = result->dim_size(i); in HandleReduce()505 context->input(0).tensor<T, NDIM>(), reduce_dim, reshape_dim); in HandleReduce()
387 const int64 reshape_dim = MathUtil::IPow(dim, count); in StrideOrInflate() local388 reshape.push_back(reshape_dim); in StrideOrInflate()393 (dim > 1 && count > 1) ? (reshape_dim - 1) / (dim - 1) : 1; in StrideOrInflate()