Searched refs:kernel_info_ (Results 1 – 6 of 6) sorted by relevance
38 kernel_info_ = dynamic_cast<KernelInfo *>(kernel_->kernel_info()); in Init()48 MS_EXCEPTION_IF_NULL(kernel_info_); in Init()49 for (auto &output_address : kernel_info_->output_address_list()) { in Init()56 for (auto &workspace_address : kernel_info_->workspace_address_list()) { in Init()344 MS_EXCEPTION_IF_NULL(kernel_info_); in FetchOutputDeviceTensor()345 auto &output_addresses = kernel_info_->output_address_list(); in FetchOutputDeviceTensor()346 const auto &kernel_mod = kernel_info_->kernel_mod(); in FetchOutputDeviceTensor()
119 MS_EXCEPTION_IF_NULL(kernel_info_); in Init()120 for (size_t i = 0; i < kernel_info_->output_address_list().size(); ++i) { in Init()126 MS_EXCEPTION_IF_NULL(kernel_info_); in FillDataBuffer()129 for (auto &device_tensor : kernel_info_->output_address_list()) { in FillDataBuffer()
51 kernel_info_(nullptr), in KernelActor()100 KernelInfo *kernel_info_; variable
113 KernelInfo *kernel_info_{nullptr};
108 kernel_info_(nullptr), in AnfNode()127 const KernelInfoDevice *kernel_info() const { return kernel_info_.get(); } in kernel_info()128 KernelInfoDevice *kernel_info() { return kernel_info_.get(); } in kernel_info()129 const KernelInfoDevicePtr &kernel_info_ptr() { return kernel_info_; } in kernel_info_ptr()130 void set_kernel_info(const KernelInfoDevicePtr &kernel_info) { kernel_info_ = kernel_info; } in set_kernel_info()227 KernelInfoDevicePtr kernel_info_; variable
545 … device_queue_ds_actor->kernel_info_ = dynamic_cast<device::KernelInfo *>((*iter)->kernel_info()); in BuildDataSourceActor()