Searched refs:add_kernels_ (Results 1 – 1 of 1) sorted by relevance
179 std::vector<std::unique_ptr<OpKernel>> add_kernels_; member in tensorflow::__anon66d29a500511::EMBenchmarkHelper196 int num_ops() const { return add_kernels_.size(); } in num_ops()235 add_kernels_.clear(); in ReInit()237 while (add_kernels_.size() < num_ops) { in ReInit()250 if (add_kernels_.empty()) { in MakeAddOp()258 add_kernels_.emplace_back(GetOpKernel(add_node_def_, &status)); in MakeAddOp()261 PrepOpKernel(add_params_.back(), add_kernels_.back().get()); in MakeAddOp()404 gpu_helper_->gpu()->Compute(add_kernels_[apc].get(), ctx.get()); in DoAddChain()