Searched refs:gpu_inputs_ (Results 1 – 1 of 1) sorted by relevance
187 gtl::InlinedVector<Tensor, 4> gpu_inputs_; member in tensorflow::__anon66d29a500511::EMBenchmarkHelper207 gpu_inputs_.clear(); in ReInit()208 while (gpu_inputs_.size() < 2) { in ReInit()209 gpu_inputs_.push_back(Tensor(gpu_helper_->gpu_allocator(), DT_FLOAT, in ReInit()312 add_inputs_[0] = TensorValue(&gpu_inputs_[0]); in PrepOpKernel()313 add_inputs_[1] = TensorValue(&gpu_inputs_[1]); in PrepOpKernel()376 Tensor alias0(gpu_inputs_[0]); in DoAddChain()377 Tensor alias1(gpu_inputs_[1]); in DoAddChain()386 se::DeviceMemoryBase gpu_dst_ptr0(DMAHelper::base(&gpu_inputs_[0]), in DoAddChain()390 se::DeviceMemoryBase gpu_dst_ptr1(DMAHelper::base(&gpu_inputs_[1]), in DoAddChain()