| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/st/python/ |
| D | test_inference_cloud.py | 24 def prepare_inputs_data(inputs, file_path, input_shapes): argument
66 def model_common_predict(context, model_path, in_data_path, input_shapes): argument
83 def runner_common_predict(context, model_path, in_data_path, input_shapes): argument
97 def test_model_inference_cpu(model_path, in_data_path, input_shapes): argument
106 def test_model_inference_gpu(model_path, in_data_path, input_shapes): argument
116 def test_model_inference_ascend(model_path, in_data_path, input_shapes): argument
126 def test_parallel_inference_cpu(model_path, in_data_path, input_shapes): argument
136 def test_parallel_inference_gpu(model_path, in_data_path, input_shapes): argument
147 def test_parallel_inference_ascend(model_path, in_data_path, input_shapes): argument
157 def test_model_group_for_ascend(model_path, in_data_path, input_shapes): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/st/scripts/experimental/ |
| D | model_info.py | 108 def convert_cmd(self, input_shapes="", output_name=""): argument
122 def benchmark_accuracy_cmd(self, input_shapes="", acc_threshold=0.5): argument
134 def benchmark_performance_cmd(self, input_shapes="", warmup_loop=3, loop=10, num_threads=2): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | stack_tests.cc | 36 std::vector<int> input_shapes[INPUT_NUM] = {{10}, {10}}; in TEST_F() local
53 std::vector<int> input_shapes[INPUT_NUM] = {{3, 4}, {3, 4}}; in TEST_F() local
72 std::vector<int> input_shapes[INPUT_NUM] = {{3, 4, 5}, {3, 4, 5}}; in TEST_F() local
91 std::vector<int> input_shapes[INPUT_NUM] = {{3, 4, 5}, {3, 4, 5}}; in TEST_F() local
110 std::vector<int> input_shapes[INPUT_NUM] = {{1, 96}, {1, 96}}; in TEST_F() local
129 std::vector<int> input_shapes[INPUT_NUM] = {{3, 4, 6}, {3, 4, 6}}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/bisheng/ |
| D | wkv_grad_bisheng_kernel.cc | 53 const auto &input_shapes = args.input_shapes; in WKVGradWorkSpaceFunc() local
66 const auto &input_shapes = args.input_shapes; in WKVGradTilingFunc() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/ |
| D | calib_data_generator.cc | 40 std::vector<std::vector<int64_t>> input_shapes; in GetInputShapes() local
105 const std::vector<std::vector<int64_t>> &input_shapes) { in DumpKernelsData()
289 auto input_shapes = GetInputShapes(graph_inputs); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/convert/ |
| D | runtime_convert.cc | 26 std::vector<std::vector<int64_t>> *input_shapes) { in ParseShapeStrToShapeMap()
78 std::vector<std::vector<int64_t>> input_shapes; in UpdateDynamicInputShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | isclose_gpu_kernel.cc | 124 std::vector<std::vector<int64_t>> input_shapes; in Init() local
161 std::vector<std::vector<int64_t>> input_shapes; in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/ascend_ge/ |
| D | ge_dynamic_utils.cc | 22 bool GeDynamicUtils::IsDynamicInputShapes(const std::vector<ShapeVector> &input_shapes) { in IsDynamicInputShapes()
28 …Utils::IsDynamicInputShapes(const std::vector<std::pair<std::string, ShapeVector>> &input_shapes) { in IsDynamicInputShapes()
76 std::vector<GeDynamicShapeInfo> input_shapes; in GetGraphInputShapes() local
414 std::vector<GeDynamicShapeInfo> input_shapes; in GetGraphOneRealShapes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | batchnorm_fp32_coder.cc | 30 std::vector<int> input_shapes = input_tensor_->shape(); in Init() local
|
| D | groupnorm_fp32_coder.cc | 30 std::vector<int> input_shapes = input_tensor_->shape(); in Init() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | rpc_recv.cc | 37 std::vector<abstract::BaseShapePtr> input_shapes; in InferShape() local
|
| D | rpc_send.cc | 37 std::vector<abstract::BaseShapePtr> input_shapes; in InferShape() local
|
| D | einsum.cc | 59 const std::vector<std::vector<int64_t>> &input_shapes, in seg_left_equation()
179 const std::vector<std::vector<int64_t>> &input_shapes, in element_map_shape()
277 std::vector<std::vector<int64_t>> input_shapes; in EinsumInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | launch_mul.cc | 34 std::vector<std::vector<int64_t>> input_shapes = {{shape}, {1}}; in ObtainMulKernelGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | single_kernel_graph.cc | 24 …ng &op_name, const std::vector<TypeId> &input_dtypes, const std::vector<ShapeVector> &input_shapes, in ConstructKernelGraphBasedOnSingleOp()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/ |
| D | node_out_shapes.cc | 32 std::vector<ShapeVector> input_shapes; in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | flops_collection.cc | 59 int64_t CalFAFlops(const std::vector<ShapeVector> &input_shapes) { in CalFAFlops()
89 int64_t CalBMMFlops(const std::vector<ShapeVector> &input_shapes, bool transpose_b) { in CalBMMFlops()
125 int64_t CalConv2DFlops(const std::vector<ShapeVector> &input_shapes, const ShapeVector &out_shape, … in CalConv2DFlops()
164 int64_t CalMatMulFlops(const std::vector<ShapeVector> &input_shapes, bool transpose_b) { in CalMatMulFlops()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/cuda/ |
| D | cuda_kernel.cc | 44 std::vector<std::vector<size_t>> input_shapes; in ReSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/ |
| D | softsign_helper.h | 36 int CalMemSize(const std::vector<std::vector<int64_t>> &input_shapes, in CalMemSize()
|
| D | mish_helper.h | 36 int CalMemSize(const std::vector<std::vector<int64_t>> &input_shapes, in CalMemSize()
|
| D | assign_helper.h | 41 int CalMemSize(const std::vector<std::vector<int64_t>> &input_shapes, in CalMemSize()
|
| D | population_count_helper.h | 36 int CalMemSize(const std::vector<std::vector<int64_t>> &input_shapes, in CalMemSize()
|
| D | erfinv_helper.h | 36 int CalMemSize(const std::vector<std::vector<int64_t>> &input_shapes, in CalMemSize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | arithmetic_info.cc | 42 Shapes input_shapes; in InferExpandShape() local
85 Shapes input_shapes = InferExpandShape(); in BaseCheckStrategy() local
371 Shapes input_shapes = InferExpandShape(); in CheckInputLayout() local
401 Shapes input_shapes = InferExpandShape(); in InferOutputLayout() local
501 Shapes input_shapes = InferExpandShape(); in CheckStrategy() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | tensor_copy_gpu_kernel.cc | 34 auto input_shapes = inputs.at(kIndex0)->GetShapeVector(); in Resize() local
|