| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | diag_impl.cu | 36 __global__ void DiagKernel(const DataType *input_ptr, DataType *output_ptr, size_t input_size, size… in DiagKernel()
48 cudaError_t CalDiag(const DataType *input_ptr, DataType *output_ptr, size_t input_size, size_t outp… in CalDiag()
|
| D | padding_impl.cu | 27 T *output_ptr) { in CalculatePaddingKernel()
34 …CalculatePadding(const T *input_ptr, size_t output_outer_size_, size_t pad_dim_size, T *output_ptr, in CalculatePadding()
|
| D | matrix_band_part_impl.cu | 27 const size_t m, const size_t n, T *output_ptr) { in MatrixBandPartDiagonalKernel()
39 const int upper, T *output_ptr) { in MatrixBandPartKernel()
58 … const size_t n, const LU *lower_ptr, const LU *upper_ptr, T *output_ptr) { in MatrixBandPartKernelBroadcast()
89 … const int64_t lower, const int64_t upper, T *output_ptr, const uint32_t &device_id, in MatrixBandPart()
111 … const size_t n, const LU *lower_ptr, const LU *upper_ptr, T *output_ptr, in MatrixBandPartBroadcast()
|
| D | as_strided_impl.cu | 23 …StridedKernel(const size_t input_size, const DataType *input_ptr, DataType *output_ptr, size_t ndi… in AsStridedKernel()
38 cudaError_t CalAsStrided(const size_t &input_size, const DataType *input_ptr, DataType *output_ptr, in CalAsStrided()
|
| D | transform_output_iterator.cuh | 32 OutputType *output_ptr; member
60 OutputType *output_ptr; member in TransformOutputIterator
66 …__host__ __device__ __forceinline__ TransformOutputIterator(T *output_ptr, ConversionOp conversion… in TransformOutputIterator()
|
| D | adaptive_avg_pool2d_impl.cu | 34 T *output_ptr = output_data + c * output_height * output_width; in AdaptiveAvgPool2DKernel() local
65 float *output_ptr = output_data + c * output_height * output_width; in AdaptiveAvgPool2DKernel() local
96 half *output_ptr = output_data + c * output_height * output_width; in AdaptiveAvgPool2DKernel() local
127 double *output_ptr = output_data + c * output_height * output_width; in AdaptiveAvgPool2DKernel() local
|
| D | masked_select_impl.cu | 35 __global__ void MaskedSelectKernel(const T *input_ptr, const size_t *index_ptr, T *output_ptr, size… in MaskedSelectKernel()
49 … T *input_broadcast_ptr, bool *mask_broadcast_ptr, T *output_ptr, size_t device_id, in MaskedSelect()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | range_fp32.h | 22 void Range(float *output_ptr, float start, float delta, int nums) { in Range()
28 void RangeInt(int32_t *output_ptr, int start, int delta, int nums) { in RangeInt()
|
| D | l2_norm_fp32.c | 30 int ThreadDivSqrtSum(const float *input_ptr, float *output_ptr, const L2NormParameter *param, const… in ThreadDivSqrtSum()
51 int ThreadTrailingAxis(const float *input_ptr, float *output_ptr, const L2NormParameter *param, con… in ThreadTrailingAxis()
|
| D | gru_fp32.c | 125 float *output_ptr = output + real_t * gru_param->output_step_; in GruUnidirectional() local
142 float *output_ptr = output + t * gru_param->output_step_; in Gru() local
162 float *output_ptr = backward_output + t * gru_param->output_step_; in Gru() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | arithmetic_grad_fp32_tests.cc | 219 float *output_ptr = reinterpret_cast<float *>(outputs[1]->MutableData()); in TEST_F() local
263 float *output_ptr = reinterpret_cast<float *>(outputs[0]->MutableData()); in TEST_F() local
309 float *output_ptr = reinterpret_cast<float *>(outputs[0]->MutableData()); in TEST_F() local
356 float *output_ptr = reinterpret_cast<float *>(outputs[1]->MutableData()); in TEST_F() local
403 float *output_ptr = reinterpret_cast<float *>(outputs[0]->MutableData()); in TEST_F() local
457 float *output_ptr = reinterpret_cast<float *>(outputs[1]->MutableData()); in TEST_F() local
502 float *output_ptr = reinterpret_cast<float *>(outputs[0]->MutableData()); in TEST_F() local
548 float *output_ptr = reinterpret_cast<float *>(outputs[1]->MutableData()); in TEST_F() local
594 float *output_ptr = reinterpret_cast<float *>(outputs[0]->MutableData()); in TEST_F() local
640 float *output_ptr = reinterpret_cast<float *>(outputs[1]->MutableData()); in TEST_F() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/ |
| D | softmax_grad_utils.c | 40 static void SoftMaxP1Simple(const float *input_ptr, float *output_ptr, float *sum_data, int start, … in SoftMaxP1Simple()
59 void SoftMaxP1(const float *input_ptr, float *output_ptr, float *sum_data, int start, int count, in… in SoftMaxP1()
89 void SoftMaxP2(const float *input_ptr, float *output_ptr, const float *sum_data, int start, int cou… in SoftMaxP2()
|
| D | dropout_grad.c | 19 void DropoutGrad(const float *yt_ptr, const float *mask, float *output_ptr, int length, float scale… in DropoutGrad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | log_softmax.c | 33 void *output_ptr = log_softmax->softmax_.base_.out_[OUTPUT_INDEX]->data_; in LogSoftmaxLastAxisRun() local
89 void *output_ptr = self->out_[OUTPUT_INDEX]->data_; in LogSoftmaxCompute() local
|
| D | pooling.c | 34 float16_t *output_ptr = (float16_t *)pooling->base_.out_[OUTPUT_INDEX]->data_; in PoolingF16RunImpl() local
55 float *output_ptr = (float *)pooling->base_.out_[OUTPUT_INDEX]->data_; in PoolingRunImpl() local
|
| D | softmax.c | 44 void *output_ptr = softmax->base_.out_[OUTPUT_INDEX]->data_; in SoftmaxLastAxisRun() local
127 void *output_ptr = self->out_[OUTPUT_INDEX]->data_; in SoftmaxCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | pad_v3_cpu_kernel.cc | 133 void PadV3CpuKernelMod::ConstantModeCompute(T *input_ptr, T *output_ptr, T constant_values) { in ConstantModeCompute()
214 void PadV3CpuKernelMod::OtherModeCompute(T *input_ptr, T *output_ptr, int64_t p) const { in OtherModeCompute()
225 void PadV3CpuKernelMod::OtherModeCompute1D(T *input_ptr, T *output_ptr, int64_t p) const { in OtherModeCompute1D()
242 void PadV3CpuKernelMod::OtherModeCompute2D(T *input_ptr, T *output_ptr, int64_t p) const { in OtherModeCompute2D()
268 void PadV3CpuKernelMod::OtherModeCompute3D(T *input_ptr, T *output_ptr, int64_t p) const { in OtherModeCompute3D()
337 auto output_ptr = static_cast<T *>(outputs[0]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | gru_fp16.c | 107 float16_t *output_ptr = output + real_t * gru_param->output_step_; in GruUnidirectionalFp16() local
123 float16_t *output_ptr = output + t * gru_param->output_step_; in GruFp16() local
142 float16_t *output_ptr = backward_output + t * gru_param->output_step_; in GruFp16() local
|
| D | range_fp16.h | 21 void RangeFp16(float16_t *output_ptr, float16_t start, float16_t delta, int nums) { in RangeFp16()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16_grad/ |
| D | dropout_grad.c | 19 void DropoutFp16Grad(const float16_t *yt_ptr, const float16_t *mask, float16_t *output_ptr, int len… in DropoutFp16Grad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | parameterized_truncated_normal_cpu_kernel.cc | 142 T **output_ptr) { in Generate()
171 … const T stddev, const T mean, T **output_ptr) { in GenerateCase1()
202 … const T stddev, const T mean, T **output_ptr) { in GenerateCase2()
239 … const T stddev, const T mean, T **output_ptr) { in GenerateCase3()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/ |
| D | tanh_int8.c | 22 void TanhInt8(const int8_t *input_ptr, int8_t *output_ptr, int size, const TanhQuantParameter *quan… in TanhInt8()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/const_fold/ |
| D | rsqrt_fp32.cc | 36 float *output_ptr = reinterpret_cast<float *>(out_tensors_.at(0)->data()); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/ |
| D | space_to_batch_int8.cc | 33 auto output_ptr = reinterpret_cast<int8_t *>(output_tensor->data()); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | invert_permutation_fp32.cc | 59 auto output_ptr = reinterpret_cast<int32_t *>(out_tensor->data()); in Run() local
|