Searched refs:result_tensor_ (Results 1 – 2 of 2) sorted by relevance
1487 result_tensor_(nullptr), in RdmaTensorRequest()1496 Tensor val = std::move(*result_tensor_); in Done()1519 if (result_tensor_ != nullptr) { in DeallocateTensors()1520 delete result_tensor_; in DeallocateTensors()1521 result_tensor_ = nullptr; in DeallocateTensors()1530 result_tensor_ = in AllocateTensors()1534 size_t tensor_size = result_tensor_->TotalBytes(); in AllocateTensors()1535 bool can_memcpy = DataTypeCanUseMemcpy(result_tensor_->dtype()); in AllocateTensors()1540 rdma_addr_ = DMAHelper::base(result_tensor_); in AllocateTensors()1547 result_tensor_->dtype(), result_tensor_->shape()); in AllocateTensors()[all …]
296 Tensor* result_tensor_; variable