| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/base/ |
| D | dtype_cast_coder.cc | 61 const int data_num) { in CastToFloat32()
74 const int data_num) { in CastToFloat16()
96 const int data_num) { in CastToOtherType()
110 int data_num = MSMIN(stride_, data_num_ - kDefaultTaskId * stride_); in DoCode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | cast.c | 27 int CastToFp32(const TensorC *input, TensorC *output, int offset, int data_num) { in CastToFp32()
56 int CastToFp16(const TensorC *input, TensorC *output, int offset, int data_num) { in CastToFp16()
89 int CastToOthers(const TensorC *input, TensorC *output, int offset, int data_num) { in CastToOthers()
132 int data_num = MSMIN(stride, cast->data_num_ - task_id * stride); in CastLaunch() local
169 int data_num = GetElementNum(in_tensor); in cast_resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | cast_fp32.cc | 66 …tCPUKernel::CastToFp32(const lite::Tensor *input, lite::Tensor *output, int offset, int data_num) { in CastToFp32()
102 …tCPUKernel::CastToFp16(const lite::Tensor *input, lite::Tensor *output, int offset, int data_num) { in CastToFp16()
140 …PUKernel::CastToOthers(const lite::Tensor *input, lite::Tensor *output, int offset, int data_num) { in CastToOthers()
188 int data_num = MSMIN(stride_, data_num_ - thread_id * stride_); in DoCast() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | reciprocal.cc | 54 uint64_t data_num = x->NumElements(); in Compute() local
85 uint32_t ReciprocalCpuKernel::ReciprocalCompute(Tensor *x, Tensor *y, uint64_t data_num, CpuKernelC… in ReciprocalCompute()
129 uint32_t ReciprocalCpuKernel::ReciprocalComputeComplex(Tensor *x, Tensor *y, uint64_t data_num, Cpu… in ReciprocalComputeComplex()
|
| D | reciprocal_grad.cc | 55 uint64_t data_num = y->NumElements(); in Compute() local
86 …ReciprocalGradCpuKernel::ReciprocalGradCompute(Tensor *y, Tensor *dy, Tensor *z, uint64_t data_num, in ReciprocalGradCompute()
122 …calGradCpuKernel::ReciprocalGradComputeComplex(Tensor *y, Tensor *dy, Tensor *z, uint64_t data_num, in ReciprocalGradComputeComplex()
|
| D | heaviside.cc | 96 const int64_t &data_num) { in NoBcastComputeParallel()
141 const int64_t &data_num) { in NoBcastComputeSingle()
171 int64_t data_num = ctx.Output(0)->NumElements(); in NoBcastCompute() local
197 int64_t data_num = ctx.Output(0)->NumElements(); in BcastCompute() local
|
| D | hypot.cc | 85 const int64_t &data_num) { in NoBcastComputeParallel()
130 const int64_t &data_num) { in NoBcastComputeSingle()
161 int64_t data_num = ctx.Output(0)->NumElements(); in NoBcastCompute() local
192 int64_t data_num = ctx.Output(0)->NumElements(); in BcastCompute() local
|
| D | expm1.cc | 91 size_t data_num = ctx.Input(0)->NumElements(); in Expm1Compute() local
121 size_t data_num = ctx.Input(0)->NumElements(); in Expm1Compute2() local
150 size_t data_num = ctx.Input(0)->NumElements(); in Expm1Compute3() local
|
| D | multilabel_margin_loss_grad.cc | 58 size_t data_num = ctx.Input(1)->GetTensorShape()->NumElements(); in MultilabelMarginLossGradCheck() local
102 size_t data_num = ctx.Input(1)->GetTensorShape()->NumElements(); in MultilabelMarginLossGradCompute() local
155 size_t data_num = ctx.Input(1)->GetTensorShape()->NumElements(); in MultilabelMarginLossGradComputeFP16() local
|
| D | sign.cc | 83 int64_t data_num = ctx.Input(0)->NumElements(); in SignCompute() local
122 int64_t data_num = ctx.Input(0)->NumElements(); in SignComputeComplex() local
|
| D | log.cc | 121 size_t data_num = ctx.Input(0)->NumElements(); in LogCompute() local
155 size_t data_num = ctx.Input(0)->NumElements(); in LogCompute2() local
212 size_t data_num = ctx.Input(0)->NumElements(); in LogCompute3() local
|
| D | igamma.cc | 31 #define SWITCH_PARALLEL(ctx, SHARD, data_num, max_core_num) … argument
105 int64_t data_num = ctx.Output(0)->NumElements(); in BcastCompute() local
194 int64_t data_num = ctx.Output(0)->NumElements(); in NoBcastCompute() local
|
| D | fill.cc | 121 uint64_t data_num = dims_tensor->GetDataSize() / sizeof(T); in CalculateDims() local
145 int64_t data_num = output->NumElements(); in FillOutput() local
|
| D | log1p.cc | 83 int64_t data_num = ctx.Input(0)->NumElements(); in Log1pCompute() local
122 int64_t data_num = ctx.Input(0)->NumElements(); in Log1pComputeComplex() local
|
| D | div.cc | 197 int64_t data_num = ctx.Output(0)->NumElements(); in NoBcastComputeInt() local
226 int64_t data_num = ctx.Output(0)->NumElements(); in NoBcastCompute() local
253 int64_t data_num = ctx.Output(0)->NumElements(); in BcastComputeInt() local
307 int64_t data_num = ctx.Output(0)->NumElements(); in BcastCompute() local
|
| D | smooth_l1_loss_v2.cc | 162 int64_t data_num = predict_tensor->NumElements(); in ComputeMean() local
183 int64_t data_num = predict_tensor->NumElements(); in ComputeSum() local
241 int64_t data_num = predict_tensor->NumElements(); in ComputeNone() local
|
| D | betainc.cc | 87 int64_t max_core_num, int64_t data_num) { in SwitchParallel()
98 uint32_t RunParallel(CpuKernelContext &ctx, std::vector<T *> data_pointers, int data_num) { in RunParallel()
|
| D | acos_grad.cc | 32 int64_t data_num = 1; variable
113 int64_t data_num = ctx.Output(0)->NumElements(); in AcosGradComputeRealType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/mpi/ |
| D | mpi_export.cc | 37 …uceScatter(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num, in MPIReduceScatter()
55 …llGather(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num) { in MPIAllGather()
|
| D | mpi_interface.cc | 69 …uceScatter(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num, in MPIReduceScatter()
81 …llGather(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num) { in MPIAllGather()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/ |
| D | allreduce_impl.cc | 68 size_t data_num = data_size / sizeof(float); in Execute() local
85 size_t data_num = data_size / sizeof(float); in RingAllReduce() local
189 size_t data_num = data_size / sizeof(float); in ReduceBroadcastAllReduce() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | sparse_to_dense_infer.c | 38 int data_num = GetElementNum(input1); in SparseToDenseInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_tile_parser.cc | 39 auto data_num = tensor_proto.tensor_content().size() / sizeof(int32_t); in Parse() local
|
| D | tf_reverse_parser.cc | 49 auto data_num = tensor_proto.tensor_content().size() / sizeof(int32_t); in Parse() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | custom_masked_fill.cc | 68 auto data_num = input->ElementsNum(); in Run() local
|