Searched defs:input_data_ptr (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | split.cc | 77 T *input_data_ptr = static_cast<T *>(value_data_ptr_); in DoCompute() local
100 uint32_t SplitCpuKernel::SplitWithOneOutput(CpuKernelContext &ctx, T *input_data_ptr, in SplitWithOneOutput()
110 uint32_t SplitCpuKernel::SplitWithDimZero(CpuKernelContext &ctx, T *input_data_ptr, std::vector<T *… in SplitWithDimZero()
126 uint32_t SplitCpuKernel::SplitCompute(CpuKernelContext &ctx, T *input_data_ptr, std::vector<T *> ou… in SplitCompute()
|
| D | unpack.cc | 71 uint32_t UnpackCpuKernel::UnpackWithOneOutput(CpuKernelContext &ctx, T *input_data_ptr, in UnpackWithOneOutput()
81 uint32_t UnpackCpuKernel::UnpackWithDimZero(CpuKernelContext &ctx, T *input_data_ptr, in UnpackWithDimZero()
101 uint32_t UnpackCpuKernel::UnpackCompute(T *input_data_ptr, std::vector<T *> output_data_vec, CpuKer… in UnpackCompute()
155 T *input_data_ptr = reinterpret_cast<T *>(value_data_ptr); in DoCompute() local
|
| D | glu.cc | 65 uint32_t GluCpuKernel::SplitWithDimZero(CpuKernelContext &ctx, T *input_data_ptr, T *output_data_pt… in SplitWithDimZero()
94 uint32_t GluCpuKernel::SplitCompute(CpuKernelContext &ctx, T *input_data_ptr, T *output_data_ptr) { in SplitCompute()
141 T *input_data_ptr = reinterpret_cast<T *>(value_data_ptr_); in DoCompute() local
|
| D | mirror_pad.cc | 118 auto input_data_ptr = reinterpret_cast<T *>(ctx.Input(0)->GetData()); in DoCompute() local
133 uint32_t MirrorPadCpuKernel::MirrorPadCompute(CpuKernelContext &ctx, T *input_data_ptr, T *output_d… in MirrorPadCompute()
|
| D | adaptive_avg_pool_3d_grad.cc | 165 auto input_data_ptr = new double[output_num]; in AdaptiveAvgPool3dGradOutTemplate() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/online_fusion/ |
| D | reduce_concat_fp32.c | 108 const float *input_data_ptr = input_datas[j] + i * inner_tile * reduce_axis_size[j]; in Fp32ReduceSumConcatFusion() local
120 const float *input_data_ptr = input_datas[input_nums - 1] + i * inner_size2; in Fp32ReduceSumConcatFusion() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | glu_cpu_kernel.cc | 33 bool GLUCpuKernelMod::SplitWithDimZero(T *input_data_ptr, T *output_data_ptr) { in SplitWithDimZero()
62 bool GLUCpuKernelMod::SplitCompute(T *input_data_ptr, T *output_data_ptr) { in SplitCompute()
|
| D | adaptive_avg_pool_2d_grad_cpu_kernel.cc | 165 std::unique_ptr<double[]> input_data_ptr = std::make_unique<double[]>(output_num); in LaunchKernel() local
|
| D | adaptive_avg_pool_3d_grad_cpu_kernel.cc | 150 std::unique_ptr<double[]> input_data_ptr = std::make_unique<double[]>(output_num); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/ |
| D | addcdiv_helper.h | 69 T *input_data_ptr = nullptr; in Process() local
|
| D | addcmul_helper.h | 69 T *input_data_ptr = nullptr; in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | nuclear_norm_cpu_kernel.cc | 463 T *input_data_ptr = reinterpret_cast<T *>(inputs[0]->device_ptr()); in ComputeTensorNuclearNorm() local
|