Searched refs:cc_ctx (Results 1 – 1 of 1) sorted by relevance
124 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_NumInputs() local125 return cc_ctx->num_inputs(); in TF_NumInputs()129 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_NumOutputs() local130 return cc_ctx->num_outputs(); in TF_NumOutputs()135 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetInput() local136 if (i < 0 || i >= cc_ctx->num_inputs()) { in TF_GetInput()140 const ::tensorflow::Tensor& cc_tensor(cc_ctx->input(i)); in TF_GetInput()149 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_SetOutput() local150 if (i < 0 || i >= cc_ctx->num_inputs()) { in TF_SetOutput()159 cc_ctx->set_output(i, cc_tensor); in TF_SetOutput()[all …]