Home
last modified time | relevance | path

Searched defs:kernels (Results 1 – 25 of 89) sorted by relevance

1234

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Druntime_pass.cc43 void Nc4hw4PassReplace(std::vector<kernel::KernelExec *> *kernels, std::vector<Tensor *> *tensors, … in Nc4hw4PassReplace()
102 bool Nc4hw4PassMatch(const std::vector<kernel::KernelExec *> *kernels, size_t index) { in Nc4hw4PassMatch()
163 auto kernels = subgraph->nodes(); in RuntimePassValid() local
177 void Nc4hw4PassAct(std::vector<kernel::KernelExec *> *kernels, std::vector<Tensor *> *tensors, int … in Nc4hw4PassAct()
212 void ConvNormC4PassActIndex(std::vector<kernel::KernelExec *> *kernels, size_t index) { in ConvNormC4PassActIndex()
249 void ConvNormC4PassAct(std::vector<kernel::KernelExec *> *kernels) { in ConvNormC4PassAct()
258 STATUS DeleteRedundantTrans(std::vector<kernel::KernelExec *> *kernels, bool *changed = nullptr) { in DeleteRedundantTrans()
336 auto &kernels = sub->nodes(); in RuntimePass() local
356 auto &kernels = sub_graph->nodes(); in GraphOptimizePass() local
Dkernel_exec_util.cc83 std::set<lite::Tensor *> KernelExecUtil::AllOutTensor(const std::vector<KernelExec *> &kernels) { in AllOutTensor()
93 …vector<KernelExec *> KernelExecUtil::SubgraphInputNodes(const std::vector<KernelExec *> &kernels) { in SubgraphInputNodes()
117 …ector<KernelExec *> KernelExecUtil::SubgraphOutputNodes(const std::vector<KernelExec *> &kernels) { in SubgraphOutputNodes()
141 …or<lite::Tensor *> KernelExecUtil::SubgraphInputTensors(const std::vector<KernelExec *> &kernels) { in SubgraphInputTensors()
174 …r<lite::Tensor *> KernelExecUtil::SubgraphOutputTensors(const std::vector<KernelExec *> &kernels) { in SubgraphOutputTensors()
207 void KernelExecUtil::InitTensorInitRefCount(const std::vector<KernelExec *> &kernels) { in InitTensorInitRefCount()
230 void KernelExecUtil::FindAllInoutKernels(const std::vector<KernelExec *> &kernels) { in FindAllInoutKernels()
265 void KernelExecUtil::FindAllInoutKernelsInSubgraphKernel(const std::vector<KernelExec *> &kernels) { in FindAllInoutKernelsInSubgraphKernel()
357 const std::vector<KernelExec *> &kernels, MSKernel *kernel) { in CreateCustomSubGraph()
367 SubGraphKernel *KernelExecUtil::CreateSubGraphKernel(const std::vector<KernelExec *> &kernels, in CreateSubGraphKernel()
Dmindrt_executor.cc71 int MindrtExecutor::PrepareGraphInput(const std::vector<kernel::KernelExec *> &kernels, in PrepareGraphInput()
102 int MindrtExecutor::PrepareGraphOutput(const std::vector<kernel::KernelExec *> &kernels, in PrepareGraphOutput()
216 int MindrtExecutor::Prepare(const std::vector<kernel::KernelExec *> &kernels, const std::vector<Ten… in Prepare()
336const std::vector<kernel::KernelExec *> &kernels, const KernelCallBack &before, in Run()
Dexecutor.h30 …virtual int Prepare(const std::vector<kernel::KernelExec *> &kernels, const std::vector<Tensor *> … in Prepare()
Dexecutor.cc24 const std::vector<kernel::KernelExec *> &kernels, const KernelCallBack &before, in Run()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/
DCMakeLists.txt17 add_library(kernels OBJECT target
24 add_library(kernels OBJECT target
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_compiler/
Dsingle_graph_scheduler.cc81 std::vector<InferKernel *> *kernels) { in CreateDelegateKernel()
108 std::vector<InferKernel *> kernels; in SelectKernel() local
134 auto kernels = execution_flow_->GetKernels(); in HandleWeightForKernels() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_conv_base_parser.cc74 STATUS ParseKernels(std::vector<int64_t> *kernels, int *conv_dims, const onnx::AttributeProto &onnx… in ParseKernels()
99 STATUS ParseKernelShape(std::vector<int64_t> *kernels, int *conv_dims, const onnx::AttributeProto &… in ParseKernelShape()
184 …S OnnxConvBaseParser::ParseVecAttr(const onnx::NodeProto &onnx_node, std::vector<int64_t> *kernels, in ParseVecAttr()
Donnx_pool_parser.cc40 std::vector<int64_t> kernels; in Parse() local
119 std::vector<int64_t> kernels; in Parse() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/gpu/opencl/
Dopencl_executor.cc24const std::vector<kernel::KernelExec *> &kernels, const KernelCallBack &before, in Run()
33const std::vector<kernel::KernelExec *> &kernels, const KernelCallBack &before, in RunOrTune()
Dopencl_executor.h33 int Prepare(const std::vector<kernel::KernelExec *> &kernels, const std::vector<Tensor *> &inputs, in Prepare()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dcpu_simple_mem_plan.cc26 auto kernels = graph->execution_order(); in MemPlan() local
70 auto kernels = graph->execution_order(); in MemAssign() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/st/
Dmindrt_parallel_runtime_test.cc29 auto kernels = session->get_kernels(); in CheckRuntime() local
96 auto kernels = session->get_kernels(); in CheckRuntime2() local
Dmindrt_parallel_test.cc38 auto kernels = lite_session->get_kernels(); in CheckOffline1() local
92 auto kernels = lite_session->get_kernels(); in CheckRuntime1() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/
Druntime_pass_tests.cc36 void Nc4hw4PassConstruct(std::vector<kernel::KernelExec *> *kernels, std::vector<lite::Tensor *> *t… in Nc4hw4PassConstruct()
140 void ConvNormC4PassConstruct(std::vector<kernel::KernelExec *> *kernels, std::vector<lite::Tensor *… in ConvNormC4PassConstruct()
246 std::vector<kernel::KernelExec *> kernels; in TEST_F() local
270 std::vector<kernel::KernelExec *> kernels; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dfill_tests.cc75 std::vector<kernel::KernelExec *> kernels{fill_kernel}; in TEST_F() local
145 std::vector<kernel::KernelExec *> kernels{fill_kernel}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/
Dgpu_somas.cc56 auto &kernels = graph.execution_order(); in InitEventInfo() local
106 auto &kernels = graph.execution_order(); in InplaceNodeProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dgpu_kernel_runtime.cc376 const auto &kernels = graph->execution_order(); in IsDistributedTraining() local
387 auto &kernels = graph->execution_order(); in FetchMemUnitSize() local
579 auto &kernels = graph->execution_order(); in RefineMemSwapScheme() local
634 auto &kernels = graph->execution_order(); in InitKernelOutputAddress() local
653 auto &kernels = graph->execution_order(); in InitKernelWorkspaceAddress() local
693 auto &kernels = graph->execution_order(); in ClearKernelOutputAddress() local
718 auto &kernels = graph->execution_order(); in ClearKernelWorkspaceAddress() local
735 const auto &kernels = graph->execution_order(); in GetLastKernel() local
763 auto &kernels = graph->execution_order(); in LaunchKernelDynamic() local
905 const auto &kernels = graph->execution_order(); in RunOpLaunchKernelDynamic() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dacl_somas.cc62 auto &kernels = graph.execution_order(); in InitEventInfo() local
105 auto &kernels = graph.execution_order(); in RuntimeNodeProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/mapper/
Dmaxunpool_mapper.cc47 vector<int32_t> kernels; in Map() local
/third_party/mesa3d/src/intel/vulkan/grl/
Dgrl_cl_kernel_gen.py136 def add_kernels(kernels, cl_file, entrypoint, libs): argument
/third_party/mesa3d/src/gallium/frontends/rusticl/core/
Dprogram.rs97 kernels: Vec<String>, field
185 pub fn kernels(&self) -> &[String] { in kernels() method
231 pub kernels: HashMap<String, Arc<NirKernelBuilds>>, field
381 let mut kernels = HashSet::new(); in from_bins() localVariable
575 let mut kernels = spirv.kernels(); in build() localVariable
669 let mut kernels = HashSet::new(); in link() localVariable
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/
Dtf_deconv_parser.cc59 std::vector<int64_t> kernels(4); in Parse() local
Dtf_conv_parser.cc57 std::vector<int64_t> kernels(4); in Parse() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/onnx/
Donnx_maxunpool_parser.cc42 std::vector<int32_t> kernels; in Parse() local

1234