Home
last modified time | relevance | path

Searched defs:input_kernel_tensors (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/device/
Dhccl_adapter_test.cc107 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(allto… in TEST_F() local
132 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(allto… in TEST_F() local
158 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(allto… in TEST_F() local
184 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(allto… in TEST_F() local
209 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(allto… in TEST_F() local
234 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(allto… in TEST_F() local
260 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(allto… in TEST_F() local
286 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(allto… in TEST_F() local
312 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(allto… in TEST_F() local
334 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(allto… in TEST_F() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Didentity.cc65 auto input_kernel_tensors = {input_x_address->kernel_tensor().get()}; in IdentityCustomizeCallWithoutContigous() local
113 auto input_kernel_tensors = {input_x_address->kernel_tensor().get()}; in IdentityCustomizeCall() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/host/
Dhost_kernel_build.cc38 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in HostOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/
Dhccl_kernel_build.cc45 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in HcclOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/rts/
Dcondition_gather.cc27 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in Init() local
Dcondition_switch.cc27 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in Init() local
Dreshape_ext.cc28 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in Init() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/
Dacl_kernel_build.cc42 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in AclOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/
Daclnn_kernel_build.cc44 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in AclnnOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/dynamic_akg/
Ddynamic_akg_gpu_kernel_build.cc39 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in SetKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_akg/
Ddynamic_akg_cpu_kernel_build.cc33 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in SetKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/
Dcpu_pass_utils.cc82 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(cast); in AddCastOpNodeToGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/bisheng/
Dbisheng_kernel_build.cc73 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(cnode… in BiShengOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dgpu_kernel_build.cc120 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(kerne… in CreateGPUKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/internal/
Dinternal_kernel_build.cc59 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in InternalKernelBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_dense_embedding_operation.cc293 std::vector<kernel::KernelTensor *> input_kernel_tensors = in LookupDeviceCache() local
354 std::vector<kernel::KernelTensor *> input_kernel_tensors = in UpdateDeviceCache() local
Ddevice_embedding_operation.cc291 const CNodePtr &kernel, const std::vector<kernel::KernelTensor *> &input_kernel_tensors, in InferOpShape()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dabstract_actor.cc120 …tor<DeviceTensor *> *const input_device_tensors, std::vector<KernelTensor *> *const input_kernel_t… in FetchInputByTensorStore()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/
Daicpu_kernel_build.cc252 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(anf_n… in AicpuOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dop_runner.cc726 std::vector<kernel::KernelTensor *> input_kernel_tensors; in GetInputKernelTensors() local
882 const auto &input_kernel_tensors = GetInputKernelTensors(input_edges); in RunSingleOpGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/
Dcpu_device_context.cc466 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(node); in CreateKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/
Dpyboost_utils.cc286 std::vector<kernel::KernelTensor *> input_kernel_tensors; in GetKernelTensorFromAddress() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_kernel_executor.cc102 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(kerne… in CreateKernelPacketKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Danf_runtime_algorithm.cc1031 std::vector<KernelTensor *> input_kernel_tensors(input_num); in GetOrCreateAllInputKernelTensors() local
1078 …std::vector<KernelTensor *> input_kernel_tensors = AnfAlgo::GetOrCreateAllInputKernelTensors(node); in GetNodeInputSizeList() local