Searched refs:allocator_attrs_ (Results 1 – 3 of 3) sorted by relevance
186 std::vector<AllocatorAttributes> allocator_attrs_; member in tensorflow::__anon9d5fcaf70511::EMBenchmarkHelper317 SetOutputAttrs(params, &allocator_attrs_); in PrepOpKernel()
511 allocator_attrs_.clear(); in CreateOpKernelContext()517 allocator_attrs_.emplace_back(attr); in CreateOpKernelContext()519 params->output_attr_array = allocator_attrs_.data(); in CreateOpKernelContext()
758 std::vector<AllocatorAttributes> allocator_attrs_; variable