Searched refs:SetInputAddr (Results 1 – 10 of 10) sorted by relevance
36 void SetInputAddr(uint8_t *input_addr) override = 0;
32 void SetInputAddr(uint8_t *input1_addr) override { input1_addr_ = input1_addr; } in SetInputAddr() function
35 void SetInputAddr(uint8_t *input_addr) override { input_addr_ = input_addr; } in SetInputAddr() function
41 void SetInputAddr(uint8_t *input_addr) override { input_addr_ = input_addr; } in SetInputAddr() function
164 launch_atomic_clean_->SetInputAddr(ar_input_addr_); in CleanAllReduceInputAddr()
281 launch_transdata_->SetInputAddr(static_cast<uint8_t *>(ptr_)); in SyncDeviceToHostAndConvertFormatBasedOnTransData()
31 void SetInputAddr(uint8_t *input1_addr) override { input1_addr_ = input1_addr; } in SetInputAddr() function
40 virtual void SetInputAddr(uint8_t *input_addr) = 0;
99 launch_mul_->SetInputAddr(ar_output_addr_); in CalculateMean()