Home
last modified time | relevance | path

Searched defs:GetOpSupport (Results 1 – 25 of 1050) sorted by relevance

12345678910>>...42

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/rl/
Dreservoir_replay_buffer_cpu_kernel.h43 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
65 std::vector<KernelAttr> GetOpSupport() { in GetOpSupport() function
87 std::vector<KernelAttr> GetOpSupport() { in GetOpSupport() function
111 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dtensors_queue_cpu_kernel.h39 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
64 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
86 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
108 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
127 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
146 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dpriority_replay_buffer_cpu_kernel.h45 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
109 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
136 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dsparse_apply_ftrl_cpu_kernel.h45 std::vector<KernelAttr> GetOpSupport() override { return OpSupport(); } in GetOpSupport() function
82 std::vector<KernelAttr> GetOpSupport() override { return OpSupport(); } in GetOpSupport() function
Dresize_nearest_neighbor_grad_cpu_kernel.h42 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dmatrix_determinant_cpu_kernel.h36 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dtrace_cpu_kernel.h43 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dsigmoid_cross_entropy_with_logits_cpu_kernel.h41 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dtrace_grad_cpu_kernel.h43 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dsigmoid_cross_entropy_with_logits_grad_cpu_kernel.h41 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dreduce_scatter_cpu_kernel.h37 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dapply_ada_max_cpu_kernel.h43 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dgrid_sampler_gpu_kernel.cc49 std::vector<KernelAttr> GridSampler2DGpuKernelMod::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::GridSampler2DGpuKernelMod
82 std::vector<KernelAttr> GridSampler3DGpuKernelMod::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::GridSampler3DGpuKernelMod
Dgrid_sampler_grad_gpu_kernel.cc55 std::vector<KernelAttr> GridSampler2DGradKernelMod::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::GridSampler2DGradKernelMod
94 std::vector<KernelAttr> GridSampler3DGradKernelMod::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::GridSampler3DGradKernelMod
Dadaptive_avg_pool2d_gpu_kernel.h39 std::vector<KernelAttr> GetOpSupport() override { return OpSupport(); } in GetOpSupport() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/rl/
Dpriority_replay_buffer_gpu_kernel.cc70 std::vector<KernelAttr> PriorityReplayBufferCreateGpuKernel::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::PriorityReplayBufferCreateGpuKernel
122 std::vector<KernelAttr> PriorityReplayBufferPushGpuKernel::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::PriorityReplayBufferPushGpuKernel
160 std::vector<KernelAttr> PriorityReplayBufferSampleGpuKernel::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::PriorityReplayBufferSampleGpuKernel
205 std::vector<KernelAttr> PriorityReplayBufferUpdateGpuKernel::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::PriorityReplayBufferUpdateGpuKernel
231 std::vector<KernelAttr> PriorityReplayBufferDestroyGpuKernel::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::PriorityReplayBufferDestroyGpuKernel
Dreservoir_replay_buffer_gpu_kernel.cc69 std::vector<KernelAttr> ReservoirReplayBufferCreateGpuKernel::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::ReservoirReplayBufferCreateGpuKernel
117 std::vector<KernelAttr> ReservoirReplayBufferPushGpuKernel::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::ReservoirReplayBufferPushGpuKernel
146 std::vector<KernelAttr> ReservoirReplayBufferSampleGpuKernel::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::ReservoirReplayBufferSampleGpuKernel
171 std::vector<KernelAttr> ReservoirReplayBufferDestroyGpuKernel::GetOpSupport() { in GetOpSupport() function in mindspore::kernel::ReservoirReplayBufferDestroyGpuKernel
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/rts/
Dcondition_switch.h33 …std::vector<KernelAttr> GetOpSupport() override { MS_LOG(EXCEPTION) << "This interface is not supp… in GetOpSupport() function
Drecv.h35 …std::vector<KernelAttr> GetOpSupport() override { MS_LOG(EXCEPTION) << "This interface is not supp… in GetOpSupport() function
Dreshape_ext.h33 …std::vector<KernelAttr> GetOpSupport() override { MS_LOG(EXCEPTION) << "This interface is not supp… in GetOpSupport() function
Dcondition_gather.h32 …std::vector<KernelAttr> GetOpSupport() override { MS_LOG(EXCEPTION) << "This interface is not supp… in GetOpSupport() function
Dsend.h34 …std::vector<KernelAttr> GetOpSupport() override { MS_LOG(EXCEPTION) << "This interface is not supp… in GetOpSupport() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/
Dsoftmax_cross_entropy_with_logits_cpu_kernel.h40 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dlog_softmax_grad_cpu_kernel.h36 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function
Dlog_softmax_cpu_kernel.h38 std::vector<KernelAttr> GetOpSupport() override { in GetOpSupport() function

12345678910>>...42