Home
last modified time | relevance | path

Searched refs:add_inputs_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc185 gtl::InlinedVector<TensorValue, 4> add_inputs_; member in tensorflow::__anon66d29a500511::EMBenchmarkHelper
198 return add_inputs_.empty() ? 0 : add_inputs_[0]->NumElements(); in tensor_size()
310 if (add_inputs_.empty()) { in PrepOpKernel()
311 add_inputs_.resize(2); in PrepOpKernel()
312 add_inputs_[0] = TensorValue(&gpu_inputs_[0]); in PrepOpKernel()
313 add_inputs_[1] = TensorValue(&gpu_inputs_[1]); in PrepOpKernel()
315 params->inputs = &add_inputs_; in PrepOpKernel()