Searched refs:dense_shape_vec (Results 1 – 1 of 1) sorted by relevance
237 ShapeVector dense_shape_vec; in InferImplMakeRowTensor() local238 (void)std::transform(std::begin(shp), std::end(shp), std::back_inserter(dense_shape_vec), in InferImplMakeRowTensor()243 if (dense_shape_vec.size() != values_shp.size()) { in InferImplMakeRowTensor()247 for (size_t i = 0; i < dense_shape_vec.size(); i++) { in InferImplMakeRowTensor()248 if (dense_shape_vec[i] < 0) { in InferImplMakeRowTensor()250 << dense_shape_vec[i]; in InferImplMakeRowTensor()253 if (i != 0 && dense_shape_vec[i] != values_shp[i]) { in InferImplMakeRowTensor()255 … << "th dimension of values " << values_shp[i] << ", but got " << dense_shape_vec[i]; in InferImplMakeRowTensor()258 auto ret = std::make_shared<AbstractRowTensor>(values->element()->BuildType(), dense_shape_vec); in InferImplMakeRowTensor()345 ShapeVector dense_shape_vec; in InferImplMakeSparseTensor() local[all …]