Searched refs:sub_ctx_ (Results 1 – 2 of 2) sorted by relevance
96 sub_ctx_.reset(new OpKernelContext(&sub_params_, 1));109 device->Compute(op, sub_ctx->sub_ctx_.get()); in ComputeBinOp()110 return sub_ctx->sub_ctx_->status(); in ComputeBinOp()
46 std::unique_ptr<OpKernelContext> sub_ctx_; variable