Searched refs:input_alloc_attrs_ (Results 1 – 7 of 7) sorted by relevance
121 input_alloc_attrs_.resize(kernel_->num_inputs()); in Init()123 for (size_t i = 0; i < input_alloc_attrs_.size(); ++i) { in Init()125 input_alloc_attrs_[i].set_on_host(host); in Init()276 params.input_alloc_attrs = input_alloc_attrs_; in Run()
254 gtl::InlinedVector<AllocatorAttributes, 4> input_alloc_attrs_; variable
242 input_alloc_attrs_.resize(total_num_inputs_); in Initialize()246 input_alloc_attrs_[output_location] = in Initialize()436 input_alloc_attrs[j] = input_alloc_attrs_[input_start_index + j]; in Run()569 input_alloc_attrs_; // Length = `total_num_inputs_`. member in tensorflow::__anon565754a20111::SingleThreadedExecutorImpl
77 return input_alloc_attrs_; in input_alloc_attrs()94 gtl::InlinedVector<AllocatorAttributes, 4> input_alloc_attrs_; variable
145 input_alloc_attrs_.resize(op_kernel_->num_inputs()); in OpKernelRunner()147 input_alloc_attrs_[i].set_on_host(input_memory_types[i] == in OpKernelRunner()
96 return input_alloc_attrs_; in input_alloc_attrs()113 gtl::InlinedVector<AllocatorAttributes, 4> input_alloc_attrs_; variable
142 input_alloc_attrs_.resize(op_kernel_->num_inputs()); in OpKernelRunner()144 input_alloc_attrs_[i].set_on_host(input_memory_types[i] == in OpKernelRunner()