Searched refs:input_addr_ (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | embedding_lookup_fp32.cc | 58 int error_code = EmbeddingLookup(input_addr_, ids_addr, output_addr, param_, task_id); in DoExcute()
78 input_addr_ = in Run()
81 if (input_addr_ == nullptr || param_->is_regulated_ == nullptr) { in Run()
97 memcpy(input_addr_ + dest_loc, input_t, sizeof(float) * in_tensors_.at(i)->ElementsNum()); in Run()
109 ms_context_->allocator->Free(input_addr_); in FreeRunBuff()
110 input_addr_ = nullptr; in FreeRunBuff()
|
| D | exp_fp32.cc | 55 …ExpFusionFp32(reinterpret_cast<float *>(input_addr_), reinterpret_cast<float *>(output_addr_), par… in DoExcute()
70 input_addr_ = in_tensors_.front()->data(); in Run()
72 CHECK_NULL_RETURN(input_addr_); in Run()
|
| D | embedding_lookup_fp32.h | 42 float *input_addr_ = nullptr; variable
|
| D | exp_fp32.h | 40 void *input_addr_ = nullptr; variable
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_launch_atomic_clean.h | 32 input_addr_(nullptr) {} in AscendLaunchAtomicClean()
35 void SetInputAddr(uint8_t *input_addr) override { input_addr_ = input_addr; } in SetInputAddr()
49 uint8_t *input_addr_; variable
|
| D | ascend_launch_transdata.h | 34 input_addr_(nullptr), in AscendLaunchTransData()
41 void SetInputAddr(uint8_t *input_addr) override { input_addr_ = input_addr; } in SetInputAddr()
55 uint8_t *input_addr_; variable
|
| D | ascend_launch_atomic_clean.cc | 57 input->addr = input_addr_; in LaunchOpKernel()
73 input_addr_ = nullptr; in FreeLaunchDeviceMem()
|
| D | ascend_launch_transdata.cc | 59 input->addr = input_addr_; in LaunchOpKernel()
75 input_addr_ = nullptr; in FreeLaunchDeviceMem()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | tile_base.cc | 129 TileSimple(input_addr_, output_addr_, begin, end, tile_parameter_); in SimpleTileImpl()
145 input_addr_ = reinterpret_cast<uint8_t *>(in_tensors_.at(0)->data()); in Run()
147 CHECK_NULL_RETURN(input_addr_); in Run()
152 Tile(input_addr_, output_addr_, reinterpret_cast<TileParameter *>(op_parameter_)); in Run()
|
| D | tile_base.h | 40 uint8_t *input_addr_ = nullptr; variable
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | exp_fp16.cc | 27 …ExpFusionFp16(reinterpret_cast<float16_t *>(input_addr_), reinterpret_cast<float16_t *>(output_add… in DoExcute()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 332 input_addr_(input_addr), in StreamSwitchTaskInfo()
339 void *input_addr() const { return input_addr_; } in input_addr()
346 void *input_addr_; variable
|