| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | cast_fp32.cc | 61 …tCPUKernel::CastToFp32(const lite::Tensor *input, lite::Tensor *output, int offset, int data_num) { in CastToFp32() argument
67 reinterpret_cast<float *>(output_data) + offset, data_num); in CastToFp32()
71 reinterpret_cast<float *>(output_data) + offset, data_num); in CastToFp32()
75 reinterpret_cast<float *>(output_data) + offset, data_num); in CastToFp32()
79 reinterpret_cast<float *>(output_data) + offset, data_num); in CastToFp32()
83 reinterpret_cast<float *>(output_data) + offset, data_num); in CastToFp32()
92 …tCPUKernel::CastToFp16(const lite::Tensor *input, lite::Tensor *output, int offset, int data_num) { in CastToFp16() argument
98 reinterpret_cast<uint16_t *>(output_data) + offset, data_num); in CastToFp16()
103 reinterpret_cast<float16_t *>(output_data) + offset, data_num); in CastToFp16()
106 reinterpret_cast<float16_t *>(output_data) + offset, data_num); in CastToFp16()
[all …]
|
| D | cast_fp32.h | 41 int CastToFp32(const lite::Tensor *input, lite::Tensor *output, int offset, int data_num);
42 int CastToFp16(const lite::Tensor *input, lite::Tensor *output, int offset, int data_num);
43 int CastToOthers(const lite::Tensor *input, lite::Tensor *output, int offset, int data_num);
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | cast_fp16.cc | 74 int data_num = MSMIN(stride_, data_num_ - thread_id * stride_); in DoCast() local
75 if (data_num <= 0) { in DoCast()
87 reinterpret_cast<int64_t *>(output_data) + offset, data_num); in DoCast()
91 reinterpret_cast<int32_t *>(output_data) + offset, data_num); in DoCast()
95 reinterpret_cast<float *>(output_data) + offset, data_num); in DoCast()
99 data_num * sizeof(float16_t)); in DoCast()
109 reinterpret_cast<int64_t *>(output_data) + offset, data_num); in DoCast()
113 reinterpret_cast<int32_t *>(output_data) + offset, data_num); in DoCast()
117 data_num * sizeof(float)); in DoCast()
121 reinterpret_cast<float16_t *>(output_data) + offset, data_num); in DoCast()
[all …]
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/ |
| D | dtype_cast_coder.cc | 38 int data_num = MSMIN(stride_, data_num_ - kDefaultTaskId * stride_); in DoCode() local
39 if (data_num <= 0) { in DoCode()
71 code << "\t\tfor (int i = 0; i < " << data_num << "; ++i) {\n"; in DoCode()
77 code.CodeFunction("Float32ToInt32", input_tensor_, output_tensor_, data_num); in DoCode()
79 code.CodeFunction("Float32ToFp16", input_tensor_, output_tensor_, data_num); in DoCode()
87 code.CodeFunction("Uint8ToFloat32", input_tensor_, output_tensor_, data_num); in DoCode()
90 code.CodeFunction("Int32ToFloat32", input_tensor_, output_tensor_, data_num); in DoCode()
93 code.CodeFunction("Fp16ToFloat32", input_tensor_, output_tensor_, data_num); in DoCode()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/mpi/ |
| D | mpi_interface.cc | 53 size_t data_num, const std::string &op_type);
58 size_t data_num);
70 …uceScatter(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num, in MPIReduceScatter() argument
73 return func(input, output, ranks_group, data_num, op_type); in MPIReduceScatter()
82 …llGather(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num) { in MPIAllGather() argument
84 return func(input, output, ranks_group, data_num); in MPIAllGather()
|
| D | mpi_export.cc | 37 …uceScatter(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num, in MPIReduceScatter() argument
43 return inst->ReduceScatter(input, output, ranks_group, data_num, op_type); in MPIReduceScatter()
55 …llGather(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num) { in MPIAllGather() argument
60 return inst->AllGather(input, output, ranks_group, data_num); in MPIAllGather()
|
| D | mpi_adapter.cc | 163 …uceScatter(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num, in ReduceScatter() argument
181 receive_count[i] = static_cast<int>(data_num); in ReduceScatter()
250 …llGather(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num) { in AllGather() argument
265 …MPI_Allgather(input, static_cast<int>(data_num), MPI_FLOAT, output, static_cast<int>(data_num), MP… in AllGather()
|
| D | mpi_interface.h | 24 …uceScatter(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num,
29 …AllGather(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num);
|
| D | mpi_export.h | 29 size_t data_num, const std::string &op_type);
32 …AllGather(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num);
|
| D | mpi_adapter.h | 39 size_t data_num, const std::string &op_type);
42 …AllGather(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | sparse_to_dense_infer.c | 38 int data_num = GetElementNum(input1); in SparseToDenseInferShape() local
39 if (input1_data == 0 || data_num > MAX_SHAPE_SIZE) { in SparseToDenseInferShape()
44 for (int i = 0; i < data_num; i++) { in SparseToDenseInferShape()
|
| D | tile_infer.c | 66 int data_num = GetElementNum(inputs[1]); in TileInferShape() local
67 multiples_size = (size_t)(data_num); in TileInferShape()
75 for (int i = 0; i < data_num; i++) { in TileInferShape()
|
| /third_party/mindspore/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
40 for (size_t i = 0; i < data_num; ++i) { in Parse()
|
| D | tf_reverse_parser.cc | 49 auto data_num = tensor_proto.tensor_content().size() / sizeof(int32_t); in Parse() local
51 for (size_t i = 0; i < data_num; ++i) { in Parse()
|
| D | tf_slice_parser.cc | 47 auto data_num = tensor_proto.tensor_content().size() / sizeof(int32_t); in Parse() local
49 for (size_t i = 0; i < data_num; ++i) { in Parse()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | lin_space_fission.cc | 60 size_t data_num = LongToSize(assist_num); in CreateTensor() local
61 for (size_t i = 0; i < data_num; ++i) { in CreateTensor()
65 auto elem_num = data_num * kFloat32Len; in CreateTensor()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | minimal_filtering_generator.c | 36 int data_num = (degree + 1) * (degree + 1); in DiagonalPlusMatrix() local
37 memset(diagonal_matrix, 0, data_num * sizeof(float)); in DiagonalPlusMatrix()
45 diagonal_matrix[data_num - 1] = 1; in DiagonalPlusMatrix()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _inner_ops.py | 1083 data_num = len(indices_shape)
1085 validator.check("size of indices", indices_num, 'size of data', data_num, Rel.EQ, self.name)
1097 for i in range(0, data_num):
|