Searched refs:op_kernel_context (Results 1 – 6 of 6) sorted by relevance
30 ctx->op_kernel_context()->set_output(i, in Compile()31 ctx->op_kernel_context()->input(i)); in Compile()
38 const Tensor& input = ctx->op_kernel_context()->input(0); in Compile()
49 ctx->op_kernel_context()->set_output(0, val); in Compile()
570 ctx->op_kernel_context()->set_output( in Compile()572 ctx->op_kernel_context()->input(update.input_index)); in Compile()
237 OpKernelContext* op_kernel_context() const { return context_; } in op_kernel_context() function
97 ResourceMgr* rm = ctx->op_kernel_context()->resource_manager(); in Compile()