Searched defs:cc_ctx (Results 1 – 3 of 3) sorted by relevance
99 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in GetAttrValue() local224 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_NumInputs() local229 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_NumOutputs() local235 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetInput() local250 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_SetOutput() local266 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in TF_OpKernelConstruction_Failure() local272 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_OpKernelContext_Failure() local395 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in TF_OpKernelConstruction_GetAttrString() local414 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in TF_OpKernelConstruction_GetAttrStringList() local438 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in TF_OpKernelConstruction_HasAttr() local[all …]
125 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextConcatenateShapes() local137 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextNumInputs() local145 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextGetInput() local157 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextRankKnown() local165 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextSetOutput() local238 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextDim() local261 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextSubshape() local
997 OpKernelContext* cc_ctx = reinterpret_cast<OpKernelContext*>(ctx); in set_tensor_data() local