| /third_party/mindspore/mindspore/lite/src/runtime/ |
| D | runtime_pass.cc | 25 void Nc4hw4PassReplace(std::vector<kernel::LiteKernel *> *kernels, std::vector<Tensor *> *tensors, … in Nc4hw4PassReplace()
87 bool Nc4hw4PassMatch(std::vector<kernel::LiteKernel *> *kernels, size_t index) { in Nc4hw4PassMatch()
143 auto kernels = subgraph->nodes(); in RuntimePassValid() local
156 void Nc4hw4PassAct(std::vector<kernel::LiteKernel *> *kernels, std::vector<Tensor *> *tensors, int … in Nc4hw4PassAct()
187 void ConvNormC4PassActIndex(std::vector<kernel::LiteKernel *> *kernels, size_t index) { in ConvNormC4PassActIndex()
221 void ConvNormC4PassAct(std::vector<kernel::LiteKernel *> *kernels) { in ConvNormC4PassAct()
238 auto &kernels = sub->nodes(); in RuntimePass() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ |
| D | CMakeLists.txt | 17 add_library(kernels OBJECT target
24 add_library(kernels OBJECT target
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_kernel_util.cc | 27 …d::set<lite::Tensor *> LiteKernelUtil::AllOutTensor(const std::vector<kernel::LiteKernel *> &kerne… in AllOutTensor()
37 …l::LiteKernel *> LiteKernelUtil::SubgraphInputNodes(const std::vector<kernel::LiteKernel *> &kerne… in SubgraphInputNodes()
62 const std::vector<kernel::LiteKernel *> &kernels) { in SubgraphOutputNodes()
86 …ite::Tensor *> LiteKernelUtil::SubgraphInputTensors(const std::vector<kernel::LiteKernel *> &kerne… in SubgraphInputTensors()
119 …te::Tensor *> LiteKernelUtil::SubgraphOutputTensors(const std::vector<kernel::LiteKernel *> &kerne… in SubgraphOutputTensors()
154 int LiteKernelUtil::TopologicalSortKernels(std::vector<kernel::LiteKernel *> *kernels) { in TopologicalSortKernels()
190 void LiteKernelUtil::InitTensorInitRefCount(const std::vector<kernel::LiteKernel *> &kernels) { in InitTensorInitRefCount()
239 void LiteKernelUtil::FindAllInoutKernels(const std::vector<kernel::LiteKernel *> &kernels) { in FindAllInoutKernels()
|
| D | mindrt_executor.cc | 25 int MindrtExecutor::PrepareInputData(const std::vector<kernel::LiteKernel *> &kernels, in PrepareInputData()
50 int MindrtExecutor::PrepareOutputData(const std::vector<kernel::LiteKernel *> &kernels, in PrepareOutputData()
95 int MindrtExecutor::Prepare(const std::vector<kernel::LiteKernel *> &kernels, const std::vector<Ten… in Prepare()
172 … const std::vector<kernel::LiteKernel *> &kernels, const KernelCallBack &before, in Run()
|
| D | executor.h | 31 …virtual int Prepare(const std::vector<kernel::LiteKernel *> &kernels, const std::vector<Tensor *> … in Prepare()
|
| D | executor.cc | 24 const std::vector<kernel::LiteKernel *> &kernels, const KernelCallBack &before, in Run()
|
| D | scheduler.cc | 69 … const std::vector<kernel::LiteKernel *> &kernels, kernel::Kernel *kernel) { in CreateCustomSubGraph()
388 std::vector<kernel::Kernel *> kernels; in ReplaceDelegateKernels() local
1109 kernel::SubGraphKernel *CreateSubGraphKernel(const std::vector<kernel::LiteKernel *> &kernels, in CreateSubGraphKernel()
1302 std::vector<kernel::LiteKernel *> kernels; in ScheduleMainSubGraphToKernels() local
1321 std::vector<kernel::LiteKernel *> kernels; in SchedulePartialToSubGraphKernel() local
1400 auto kernels = ScheduleSubGraphToSubGraphKernels(cur_subgraph_index); in ScheduleGraphToKernels() local
1638 kernel::SubGraphType Scheduler::PartialSubGraphType(const std::vector<kernel::LiteKernel *> &kernel… in PartialSubGraphType()
1723 bool Scheduler::IsControlFlowParttern(const std::vector<kernel::LiteKernel *> &kernels) { in IsControlFlowParttern()
1735 int Scheduler::ConstructControlFlowMainGraph(std::vector<kernel::LiteKernel *> *kernels) { in ConstructControlFlowMainGraph()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_conv_base_parser.cc | 53 STATUS ParseKernels(std::vector<int64_t> *kernels, bool *conv1d, const onnx::AttributeProto &onnx_n… in ParseKernels()
72 STATUS ParseKernelShape(std::vector<int64_t> *kernels, bool *conv1d, const onnx::AttributeProto &on… in ParseKernelShape()
133 …S OnnxConvBaseParser::ParseVecAttr(const onnx::NodeProto &onnx_node, std::vector<int64_t> *kernels, in ParseVecAttr()
|
| D | onnx_pool_parser.cc | 33 std::vector<int64_t> kernels; in Parse() local
106 std::vector<int64_t> kernels; in Parse() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/gpu/opencl/ |
| D | opencl_executor.cc | 24 … const std::vector<kernel::LiteKernel *> &kernels, const KernelCallBack &before, in Run()
33 … const std::vector<kernel::LiteKernel *> &kernels, const KernelCallBack &before, in RunOrTune()
|
| D | opencl_executor.h | 34 int Prepare(const std::vector<kernel::LiteKernel *> &kernels, const std::vector<Tensor *> &inputs, in Prepare()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_simple_mem_plan.cc | 25 auto kernels = graph->execution_order(); in MemPlan() local
69 auto kernels = graph->execution_order(); in MemAssign() local
|
| /third_party/mindspore/mindspore/lite/test/st/ |
| D | mindrt_parallel_runtime_test.cc | 30 auto kernels = lite_session->get_kernels(); in CheckRuntime() local
92 auto kernels = lite_session->get_kernels(); in CheckRuntime2() local
|
| D | mindrt_parallel_test.cc | 35 auto kernels = lite_session->get_kernels(); in CheckOffline1() local
89 auto kernels = lite_session->get_kernels(); in CheckRuntime1() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/ |
| D | runtime_pass_tests.cc | 33 void Nc4hw4PassConstruct(std::vector<kernel::LiteKernel *> *kernels, std::vector<lite::Tensor *> *t… in Nc4hw4PassConstruct()
119 void ConvNormC4PassConstruct(std::vector<kernel::LiteKernel *> *kernels, std::vector<lite::Tensor *… in ConvNormC4PassConstruct()
206 std::vector<kernel::LiteKernel *> kernels; in TEST_F() local
230 std::vector<kernel::LiteKernel *> kernels; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_build.cc | 30 void CreateGPUKernel(const std::vector<CNodePtr> &kernels) { in CreateGPUKernel()
|
| D | gpu_kernel_runtime.cc | 360 const auto &kernels = graph->execution_order(); in IsDistributedTraining() local
371 auto &kernels = graph->execution_order(); in FetchMemUnitSize() local
562 auto &kernels = graph->execution_order(); in RefineMemSwapScheme() local
617 auto &kernels = graph->execution_order(); in InitKernelOutputAddress() local
636 auto &kernels = graph->execution_order(); in InitKernelWorkspaceAddress() local
676 auto &kernels = graph->execution_order(); in ClearKernelOutputAddress() local
701 auto &kernels = graph->execution_order(); in ClearKernelWorkspaceAddress() local
718 const auto &kernels = graph->execution_order(); in GetLastKernel() local
745 auto &kernels = graph->execution_order(); in LaunchKernelDynamic() local
874 const auto &kernels = graph->execution_order(); in RunOpLaunchKernelDynamic() local
[all …]
|
| D | gpu_launch_kernel.cc | 56 auto kernels = kernel_graph->execution_order(); in KernelBuild() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | fill_tests.cc | 76 std::vector<kernel::LiteKernel *> kernels{fill_kernel}; in TEST_F() local
146 std::vector<kernel::LiteKernel *> kernels{fill_kernel}; in TEST_F() local
|
| D | cast_tests.cc | 103 std::vector<kernel::LiteKernel *> kernels{cast_kernel}; in TEST_F() local
212 std::vector<kernel::LiteKernel *> kernels{cast_kernel}; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_deconv_parser.cc | 54 std::vector<int64_t> kernels(4); in Parse() local
|
| D | tf_conv_parser.cc | 54 std::vector<int64_t> kernels(4); in Parse() local
|
| /third_party/mindspore/config/ |
| D | data_dump.json | 8 "kernels": ["Default/Conv-op12"], array
|
| /third_party/mindspore/mindspore/lite/test/ut/src/ |
| D | utils_test.cc | 58 std::vector<kernel::LiteKernel *> kernels = {kernel0.get(), kernel1.get(), kernel2.get()}; in TEST_F() local
|
| /third_party/mindspore/include/api/ |
| D | delegate.h | 39 DelegateModel(std::vector<kernel::Kernel *> *kernels, const std::vector<MSTensor> &inputs, in DelegateModel()
|