Home
last modified time | relevance | path

Searched refs:kernels (Results 1 – 25 of 267) sorted by relevance

1234567891011

/third_party/mindspore/mindspore/lite/minddata/
DCMakeLists.txt87 AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/kernels MINDDATA_KERNELS_SRC_FILES)
89 AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/kernels/image MINDDATA_KERNELS_IMAGE_SRC_FILES)
91 AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/kernels/data MINDDATA_KERNELS_DATA_SRC_FILES)
96 AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/text/kernels MINDDATA_TEXT_KERNELS_SRC_FILES)
100 AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/kernels/image/lite_cv MINDDATA_KERNELS_IMAGE_LITE_CV_FILES)
102 AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/kernels/ir MINDDATA_KERNELS_IR_SRC_FILES)
106 include_directories("${MINDDATA_DIR}/kernels/image")
120 ${MINDDATA_DIR}/kernels/data/data_utils.cc
207 ${MINDDATA_DIR}/kernels/tensor_op.cc
208 ${MINDDATA_DIR}/kernels/image/affine_op.cc
[all …]
/third_party/mindspore/mindspore/lite/tools/dataset/cropper/
Dassociations.txt1 …kernels/ir/vision/uniform_aug_ir.cc", "randomhorizontalflip": "mindspore/ccsrc/minddata/dataset/ke…
/third_party/mindspore/mindspore/lite/src/runtime/
Druntime_pass.cc25 void Nc4hw4PassReplace(std::vector<kernel::LiteKernel *> *kernels, std::vector<Tensor *> *tensors, … in Nc4hw4PassReplace() argument
26 kernel::LiteKernel *conv_kernel = kernels->at(index); in Nc4hw4PassReplace()
70 VectorErase(kernels, transpose_kernel); in Nc4hw4PassReplace()
80 VectorErase(kernels, transpose2_kernel); in Nc4hw4PassReplace()
87 bool Nc4hw4PassMatch(std::vector<kernel::LiteKernel *> *kernels, size_t index) { in Nc4hw4PassMatch() argument
88 kernel::LiteKernel *start_kernel = kernels->at(index); in Nc4hw4PassMatch()
122 auto start_iter = find(kernels->begin(), kernels->end(), start_kernel); in Nc4hw4PassMatch()
123 auto start_index = std::distance(kernels->begin(), start_iter); in Nc4hw4PassMatch()
124 auto traspose_nhwc2nchw_iter = find(kernels->begin(), kernels->end(), traspose_nhwc2nchw_kernel); in Nc4hw4PassMatch()
125 auto traspose_nhwc2nchw_index = std::distance(kernels->begin(), traspose_nhwc2nchw_iter); in Nc4hw4PassMatch()
[all …]
/third_party/mindspore/mindspore/lite/src/
Dlite_kernel_util.cc27 …et<lite::Tensor *> LiteKernelUtil::AllOutTensor(const std::vector<kernel::LiteKernel *> &kernels) { in AllOutTensor() argument
29 for (const auto &kernel_in_subgraph : kernels) { in AllOutTensor()
37 …iteKernel *> LiteKernelUtil::SubgraphInputNodes(const std::vector<kernel::LiteKernel *> &kernels) { in SubgraphInputNodes() argument
39 std::set<lite::Tensor *> all_out_tensors = AllOutTensor(kernels); in SubgraphInputNodes()
40 for (const auto &kernel : kernels) { in SubgraphInputNodes()
62 const std::vector<kernel::LiteKernel *> &kernels) { in SubgraphOutputNodes() argument
64 for (const auto &kernel : kernels) { in SubgraphOutputNodes()
69 for (const auto &kernel : kernels) { in SubgraphOutputNodes()
86 …:Tensor *> LiteKernelUtil::SubgraphInputTensors(const std::vector<kernel::LiteKernel *> &kernels) { in SubgraphInputTensors() argument
88 std::vector<kernel::LiteKernel *> input_nodes = SubgraphInputNodes(kernels); in SubgraphInputTensors()
[all …]
Dmindrt_executor.cc25 int MindrtExecutor::PrepareInputData(const std::vector<kernel::LiteKernel *> &kernels, in PrepareInputData() argument
27 for (size_t j = 0; j < kernels.size(); ++j) { in PrepareInputData()
28 auto in_tensor_size = kernels[j]->in_tensors().size(); in PrepareInputData()
30 auto tensor = kernels[j]->in_tensors()[k]; in PrepareInputData()
50 int MindrtExecutor::PrepareOutputData(const std::vector<kernel::LiteKernel *> &kernels, in PrepareOutputData() argument
67 for (size_t j = 0; j < kernels.size(); ++j) { in PrepareOutputData()
68 auto out_tensor_size = kernels[j]->out_tensors().size(); in PrepareOutputData()
70 if (subgraph_output_tensor != kernels[j]->out_tensors()[k]) { in PrepareOutputData()
95 int MindrtExecutor::Prepare(const std::vector<kernel::LiteKernel *> &kernels, const std::vector<Ten… in Prepare() argument
104 op_actors_ = CreateOpActor(kernels, ctx); in Prepare()
[all …]
Dlite_kernel_util.h27 …:vector<kernel::LiteKernel *> SubgraphInputNodes(const std::vector<kernel::LiteKernel *> &kernels);
28 …vector<kernel::LiteKernel *> SubgraphOutputNodes(const std::vector<kernel::LiteKernel *> &kernels);
29 …std::vector<lite::Tensor *> SubgraphInputTensors(const std::vector<kernel::LiteKernel *> &kernels);
30 …td::vector<lite::Tensor *> SubgraphOutputTensors(const std::vector<kernel::LiteKernel *> &kernels);
31 static int TopologicalSortKernels(std::vector<kernel::LiteKernel *> *kernels);
32 static void InitTensorInitRefCount(const std::vector<kernel::LiteKernel *> &kernels);
41 static void FindAllInoutKernels(const std::vector<kernel::LiteKernel *> &kernels);
44 static std::set<lite::Tensor *> AllOutTensor(const std::vector<kernel::LiteKernel *> &kernels);
Dmindrt_executor.h35 int Prepare(const std::vector<kernel::LiteKernel *> &kernels, const std::vector<Tensor *> &inputs,
39 const std::vector<kernel::LiteKernel *> &kernels, const KernelCallBack &before = nullptr,
50 …int PrepareInputData(const std::vector<kernel::LiteKernel *> &kernels, const std::vector<Tensor *>…
51 …int PrepareOutputData(const std::vector<kernel::LiteKernel *> &kernels, const std::vector<Tensor *…
Dscheduler.cc69 … const std::vector<kernel::LiteKernel *> &kernels, kernel::Kernel *kernel) { in CreateCustomSubGraph() argument
70 …kernel = new (std::nothrow) kernel::CustomSubGraph(input_kernels, output_kernels, kernels, kernel); in CreateCustomSubGraph()
388 std::vector<kernel::Kernel *> kernels; in ReplaceDelegateKernels() local
390 kernels.push_back((*dst_kernels)[i]->kernel()); in ReplaceDelegateKernels()
397 …new (std::nothrow) DelegateModel<schema::Primitive>(&kernels, ms_inputs_, ms_outputs_, primitives_… in ReplaceDelegateKernels()
421 for (auto kernel : kernels) { in ReplaceDelegateKernels()
1109 kernel::SubGraphKernel *CreateSubGraphKernel(const std::vector<kernel::LiteKernel *> &kernels, in CreateSubGraphKernel() argument
1121 input_tensors = kernel::LiteKernelUtil::SubgraphInputTensors(kernels); in CreateSubGraphKernel()
1126 output_tensors = kernel::LiteKernelUtil::SubgraphOutputTensors(kernels); in CreateSubGraphKernel()
1132 …::vector<kernel::LiteKernel *> input_kernels = kernel::LiteKernelUtil::SubgraphInputNodes(kernels); in CreateSubGraphKernel()
[all …]
Dexecutor.cc24 const std::vector<kernel::LiteKernel *> &kernels, const KernelCallBack &before, in Run() argument
33 for (auto *kernel : kernels) { in Run()
39 for (auto kernel : kernels) { in Run()
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_conv_base_parser.cc53 STATUS ParseKernels(std::vector<int64_t> *kernels, bool *conv1d, const onnx::AttributeProto &onnx_n… in ParseKernels() argument
54 MS_ASSERT(kernels != nullptr && conv1d != nullptr); in ParseKernels()
58 kernels->push_back(1); in ParseKernels()
59 kernels->push_back(onnx_node_attr.ints(0)); in ParseKernels()
62 kernels->push_back(onnx_node_attr.ints(0)); in ParseKernels()
63 kernels->push_back(onnx_node_attr.ints(1)); in ParseKernels()
72 STATUS ParseKernelShape(std::vector<int64_t> *kernels, bool *conv1d, const onnx::AttributeProto &on… in ParseKernelShape() argument
73 MS_ASSERT(kernels != nullptr && conv1d != nullptr); in ParseKernelShape()
77 kernels->push_back(1); in ParseKernelShape()
78 kernels->push_back(onnx_node_attr.ints(0)); in ParseKernelShape()
[all …]
Donnx_pool_parser.cc33 std::vector<int64_t> kernels; in Parse() local
40 kernels.push_back(onnx_node_attr.ints(0)); in Parse()
41 kernels.push_back(onnx_node_attr.ints(1)); in Parse()
42 prim->set_kernel_size(kernels); in Parse()
106 std::vector<int64_t> kernels; in Parse() local
113 kernels.push_back(onnx_node_attr.ints(0)); in Parse()
114 kernels.push_back(onnx_node_attr.ints(1)); in Parse()
115 prim->set_kernel_size(kernels); in Parse()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/
Druntime_pass_tests.cc24 extern void Nc4hw4PassAct(std::vector<kernel::LiteKernel *> *kernels, std::vector<Tensor *> *tensor…
25 extern void ConvNormC4PassAct(std::vector<kernel::LiteKernel *> *kernels);
33 void Nc4hw4PassConstruct(std::vector<kernel::LiteKernel *> *kernels, std::vector<lite::Tensor *> *t… in Nc4hw4PassConstruct() argument
52 kernels->push_back(conv_kernel); in Nc4hw4PassConstruct()
65 kernels->push_back(trans_kernel); in Nc4hw4PassConstruct()
78 kernels->push_back(in_kernel); in Nc4hw4PassConstruct()
91 kernels->push_back(trans2_kernel); in Nc4hw4PassConstruct()
106 kernels->push_back(conv2_kernel); in Nc4hw4PassConstruct()
119 void ConvNormC4PassConstruct(std::vector<kernel::LiteKernel *> *kernels, std::vector<lite::Tensor *… in ConvNormC4PassConstruct() argument
138 kernels->push_back(conv1_kernel); in ConvNormC4PassConstruct()
[all …]
/third_party/mesa3d/src/compiler/clc/
Dclc_helpers.cpp130 for (auto &iter : kernels) { in parseEntryPoint()
139 kernels.push_back(SPIRVKernelInfo(funcId, name)); in parseEntryPoint()
152 for (auto &kernel : kernels) { in parseFunction()
193 for (auto &kernel : kernels) { in parseName()
233 for (auto &kernel : kernels) { in parseTypePointer()
257 for (auto &kernel : kernels) { in parseOpString()
313 for (auto &kernel : kernels) { in applyDecoration()
410 for (auto &kernel : kernels) { in parseOpTypeImage()
428 for (auto& kernel : kernels) { in parseExecutionMode()
591 for (auto &kernel : kernels) { in parsingComplete()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/
DCMakeLists.txt56 include_directories(${CMAKE_SOURCE_DIR}/mindspore/ccsrc/minddata/dataset/kernels/image)
73 add_subdirectory(kernels)
82 add_dependencies(kernels-image core)
83 add_dependencies(kernels-data core)
84 add_dependencies(kernels-soft-dvpp-image core soft-dvpp-utils)
85 add_dependencies(kernels core)
94 add_dependencies(audio-kernels core)
96 add_dependencies(audio-ir-kernels core)
98 add_dependencies(text-kernels core)
100 add_dependencies(text-ir-kernels core)
[all …]
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_deconv_parser.cc54 std::vector<int64_t> kernels(4); in Parse() local
55 if (ParseKernels(*weight_node, format, &kernels) != RET_OK) { in Parse()
59 prim->set_kernel_size({kernels[0], kernels[1]}); in Parse()
60 prim->set_out_channel(kernels[2]); in Parse()
61 prim->set_in_channel(kernels[3]); in Parse()
Dtf_conv_parser.cc54 std::vector<int64_t> kernels(4); in Parse() local
55 if (ParseKernels(*weight_node, format, &kernels) != RET_OK) { in Parse()
59 prim->set_kernel_size({kernels[0], kernels[1]}); in Parse()
60 prim->set_out_channel(kernels[3]); in Parse()
61 prim->set_in_channel(kernels[2]); in Parse()
/third_party/mindspore/mindspore/lite/test/st/
Dmindrt_parallel_test.cc35 auto kernels = lite_session->get_kernels(); in CheckOffline1() local
36 if (kernels.size() != 4) { in CheckOffline1()
41 kernel::SubGraphKernel *subgraph0 = reinterpret_cast<kernel::SubGraphKernel *>(kernels[0]); in CheckOffline1()
51 kernel::SubGraphKernel *subgraph1 = reinterpret_cast<kernel::SubGraphKernel *>(kernels[1]); in CheckOffline1()
63 kernel::SubGraphKernel *subgraph2 = reinterpret_cast<kernel::SubGraphKernel *>(kernels[2]); in CheckOffline1()
75 kernel::SubGraphKernel *subgraph3 = reinterpret_cast<kernel::SubGraphKernel *>(kernels[3]); in CheckOffline1()
89 auto kernels = lite_session->get_kernels(); in CheckRuntime1() local
90 if (kernels.size() != 6) { in CheckRuntime1()
Dmindrt_parallel_runtime_test.cc30 auto kernels = lite_session->get_kernels(); in CheckRuntime() local
34 for (auto kernel : kernels) { in CheckRuntime()
43 if (kernels.size() != 6) { in CheckRuntime()
92 auto kernels = lite_session->get_kernels(); in CheckRuntime2() local
94 for (auto kernel : kernels) { in CheckRuntime2()
100 if (kernels.size() != 6) { in CheckRuntime2()
Dmultiple_device_test.cc225 void CheckResult(std::vector<mindspore::kernel::LiteKernel *> kernels, int mode) { in CheckResult() argument
235 ASSERT_EQ(1, kernels.size()); in CheckResult()
237 auto subgraph1 = reinterpret_cast<mindspore::kernel::SubGraphKernel *>(kernels.at(0)); in CheckResult()
242 ASSERT_EQ(3, kernels.size()); in CheckResult()
244 auto subgraph0 = kernels.at(0); in CheckResult()
247 auto subgraph1 = reinterpret_cast<mindspore::kernel::SubGraphKernel *>(kernels.at(1)); in CheckResult()
251 auto subgraph2 = kernels.at(2); in CheckResult()
256 ASSERT_EQ(3, kernels.size()); in CheckResult()
258 auto subgraph0 = reinterpret_cast<mindspore::kernel::SubGraphKernel *>(kernels.at(0)); in CheckResult()
262 auto subgraph1 = reinterpret_cast<mindspore::kernel::SubGraphKernel *>(kernels.at(1)); in CheckResult()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
DCMakeLists.txt8 add_library(kernels-image OBJECT
69 add_dependencies(kernels-image kernels-soft-dvpp-image kernels-dvpp-image)
71 add_dependencies(kernels-image kernels-soft-dvpp-image)
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dfill_tests.cc76 std::vector<kernel::LiteKernel *> kernels{fill_kernel}; in TEST_F() local
84 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_inner_k… in TEST_F()
146 std::vector<kernel::LiteKernel *> kernels{fill_kernel}; in TEST_F() local
154 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_inner_k… in TEST_F()
Dcast_tests.cc103 std::vector<kernel::LiteKernel *> kernels{cast_kernel}; in TEST_F() local
117 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_inner_k… in TEST_F()
212 std::vector<kernel::LiteKernel *> kernels{cast_kernel}; in TEST_F() local
227 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_inner_k… in TEST_F()
/third_party/mindspore/mindspore/lite/test/ut/src/
Dutils_test.cc58 std::vector<kernel::LiteKernel *> kernels = {kernel0.get(), kernel1.get(), kernel2.get()}; in TEST_F() local
60 auto input_kernels = kernel::LiteKernelUtil::SubgraphInputNodes(kernels); in TEST_F()
62 auto output_kernels = kernel::LiteKernelUtil::SubgraphOutputNodes(kernels); in TEST_F()
64 auto input_tensors = kernel::LiteKernelUtil::SubgraphInputTensors(kernels); in TEST_F()
66 auto output_tensors = kernel::LiteKernelUtil::SubgraphOutputTensors(kernels); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
DCMakeLists.txt5 python/bindings/dataset/audio/kernels/ir/bindings.cc
17 python/bindings/dataset/kernels/bindings.cc
18 python/bindings/dataset/kernels/ir/bindings.cc
19 python/bindings/dataset/kernels/ir/image/bindings.cc
21 python/bindings/dataset/text/kernels/ir/bindings.cc
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/
DCMakeLists.txt17 add_library(kernels OBJECT target
22 target_include_directories(kernels PRIVATE ${pybind11_INCLUDE_DIRS})
24 add_library(kernels OBJECT target

1234567891011