Searched defs:data_num (Results 1 – 13 of 13) sorted by relevance
| /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()
92 …tCPUKernel::CastToFp16(const lite::Tensor *input, lite::Tensor *output, int offset, int data_num) { in CastToFp16()
124 …PUKernel::CastToOthers(const lite::Tensor *input, lite::Tensor *output, int offset, int data_num) { in CastToOthers()
165 int data_num = MSMIN(stride_, data_num_ - thread_id * stride_); in DoCast() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/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 | 70 …uceScatter(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num, in MPIReduceScatter()
82 …llGather(const float *input, float *output, const std::vector<int> &ranks_group, size_t 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()
250 …llGather(const float *input, float *output, const std::vector<int> &ranks_group, size_t data_num) { in AllGather()
|
| /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
|
| /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
|
| D | tf_reverse_parser.cc | 49 auto data_num = tensor_proto.tensor_content().size() / sizeof(int32_t); in Parse() local
|
| D | tf_slice_parser.cc | 47 auto data_num = tensor_proto.tensor_content().size() / sizeof(int32_t); in Parse() local
|
| /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
|
| D | tile_infer.c | 66 int data_num = GetElementNum(inputs[1]); in TileInferShape() local
|
| /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
|
| /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
|
| /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
|