Searched refs:output_alloc_attrs_ (Results 1 – 2 of 2) sorted by relevance
124 output_alloc_attrs_.resize(kernel_->num_outputs()); in Init()125 for (size_t i = 0; i < output_alloc_attrs_.size(); ++i) { in Init()126 output_alloc_attrs_[i].set_on_host(kernel_->output_memory_types()[i] == in Init()255 params.output_attr_array = output_alloc_attrs_.data(); in Run()
246 gtl::InlinedVector<AllocatorAttributes, 1> output_alloc_attrs_; variable