Home
last modified time | relevance | path

Searched defs:new_shapes (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/ascend/model/
Ddyn_shape_process.cc65 bool DynShapeProcess::CheckAndGetBatchSize(const std::vector<ShapeVector> &new_shapes, int32_t *bat… in CheckAndGetBatchSize()
80 bool DynShapeProcess::CheckAndGetDynamicDims(const std::vector<ShapeVector> &new_shapes, aclmdlIODi… in CheckAndGetDynamicDims()
91 bool DynShapeProcess::CheckAndGetImageSize(const std::vector<ShapeVector> &new_shapes, int32_t *hei… in CheckAndGetImageSize()
107 bool DynShapeProcess::CheckBatchSize(const std::vector<ShapeVector> &new_shapes) { in CheckBatchSize()
137 bool DynShapeProcess::CheckDynamicDims(const std::vector<ShapeVector> &new_shapes) { in CheckDynamicDims()
161 bool DynShapeProcess::CheckImageSize(const std::vector<ShapeVector> &new_shapes) { in CheckImageSize()
208 bool DynShapeProcess::GetRealBatchSize(const std::vector<ShapeVector> &new_shapes, int32_t *batch_s… in GetRealBatchSize()
229 bool DynShapeProcess::GetRealDynamicDims(const std::vector<ShapeVector> &new_shapes, aclmdlIODims *… in GetRealDynamicDims()
249 bool DynShapeProcess::GetRealImageSize(const std::vector<ShapeVector> &new_shapes, int32_t *height_… in GetRealImageSize()
Dmodel_process.cc647 bool ModelProcess::ResetInputSize(const std::vector<ShapeVector> &new_shapes) { in ResetInputSize()
702 bool ModelProcess::Resize(const std::vector<ShapeVector> &new_shapes) { in Resize()
744 bool ModelProcess::ResizeDynamicInputShape(const std::vector<ShapeVector> &new_shapes) { in ResizeDynamicInputShape()
792 bool ModelProcess::ResizeDynamicInputShapeRange(const std::vector<ShapeVector> &new_shapes) { in ResizeDynamicInputShapeRange()
823 bool ModelProcess::ResizeDynamicBatchAndImageSize(const std::vector<ShapeVector> &new_shapes) { in ResizeDynamicBatchAndImageSize()
Dmodel_infer.cc192 bool ModelInfer::Resize(const std::vector<ShapeVector> &new_shapes) { in Resize()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/session/
Dlite_graph_executor.h59 const std::vector<std::vector<int64_t>> &new_shapes) { in Resize()
Dsingle_op_session.cc432 std::vector<ShapeVector> new_shapes; in RunGraph() local
456 Status SingleOpInferSession::OnNewInputShapes(const std::vector<ShapeVector> &new_shapes) { in OnNewInputShapes()
Ddelegate_session.cc282 const std::vector<std::vector<int64_t>> &new_shapes) { in Resize()
Dascend_native_session.cc443 const std::vector<std::vector<int64_t>> &new_shapes) { in Resize()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_subgraph.cc639 int TensorRTSubGraph::OnNewInputShapes(const std::vector<ShapeVector> &new_shapes) { in OnNewInputShapes()
774 std::vector<ShapeVector> new_shapes; in PreExecute() local
937 …TSubGraph::Resize(const std::vector<tensor::Tensor> &, const std::vector<ShapeVector> &new_shapes)… in Resize()
Dtensorrt_graph_executor.cc631 const std::vector<std::vector<int64_t>> &new_shapes) { in Resize()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dget_next_info.cc209 ValuePtr new_shapes = MakeValue(out_shapes); in InferReplaceOps() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/ascend/src/
Dcustom_ascend_kernel.cc225 std::vector<ShapeVector> new_shapes; in OnNewInputShapes() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/python/src/
Dmodel_pybind.cc53 const std::vector<std::vector<int64_t>> &new_shapes) { in PyModelResize()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parallel_compile/akg_compiler/
Dbuild_tbe_kernel.py87 def add_new_shape(names, shapes, new_shapes, inputs): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/cloud_infer/runtime_cpp/
Dmain.cc209 std::vector<std::vector<int64_t>> new_shapes; in ResizeModel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/runtime_cpp/
Dmain.cc325 std::vector<std::vector<int64_t>> new_shapes; in ResizeInputsTensorShape() local