Searched refs:tensor_shp (Results 1 – 2 of 2) sorted by relevance
132 ShapeVector tensor_shp({static_cast<int64_t>(input_shape->shape().size())}); in CreateDynamicShapeCNode() local134 … std::make_shared<abstract::AbstractTensor>(kInt64, std::make_shared<abstract::Shape>(tensor_shp)); in CreateDynamicShapeCNode()
707 ShapeVector tensor_shp({static_cast<int64_t>(shape.size())}); in InferImplDynamicShape() local712 auto tensor = std::make_shared<AbstractTensor>(elem, std::make_shared<Shape>(tensor_shp)); in InferImplDynamicShape()717 …auto tensor = std::make_shared<tensor::Tensor>(kNumberTypeInt64, tensor_shp, shape.data(), shp_buf… in InferImplDynamicShape()