Searched refs:allocator_attrs_ (Results 1 – 2 of 2) sorted by relevance
198 allocator_attrs_.clear(); in CreateOpKernelContext()204 allocator_attrs_.emplace_back(attr); in CreateOpKernelContext()206 params_->output_attr_array = gtl::vector_as_array(&allocator_attrs_); in CreateOpKernelContext()
184 std::vector<AllocatorAttributes> allocator_attrs_; variable