Searched refs:input_alloc_attrs_ (Results 1 – 3 of 3) sorted by relevance
115 input_alloc_attrs_.resize(kernel_->num_inputs()); in Init()117 for (size_t i = 0; i < input_alloc_attrs_.size(); ++i) { in Init()119 input_alloc_attrs_[i].set_on_host(host); in Init()254 params.input_alloc_attrs = &input_alloc_attrs_; in Run()
244 gtl::InlinedVector<AllocatorAttributes, 4> input_alloc_attrs_; variable
228 input_alloc_attrs_.resize(total_num_inputs_); in Initialize()232 input_alloc_attrs_[output_location] = in Initialize()407 input_alloc_attrs[j] = input_alloc_attrs_[input_start_index + j]; in Run()525 input_alloc_attrs_; // Length = `total_num_inputs_`. member in tensorflow::data::__anonb15a461d0111::SingleThreadedExecutorImpl