Searched defs:cc_ctx (Results 1 – 4 of 4) sorted by relevance
102 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in GetAttrValue() local227 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_NumInputs() local232 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_NumOutputs() local238 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetInput() local253 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_SetOutput() local269 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in TF_OpKernelConstruction_Failure() local275 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_OpKernelContext_Failure() local398 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in TF_OpKernelConstruction_GetAttrString() local417 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in TF_OpKernelConstruction_GetAttrStringList() local441 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
141 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in GetTrainingVariableMutex() local165 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_AssignVariable() local240 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_MaybeLockVariableInputMutexesInOrder() local301 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetInputTensorFromVariable() local329 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_OpKernelContext_ForwardRefInputToRefOutput() local347 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetInputByName() local367 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in TF_OpKernelConstruction_GetAttrTensorShape() local
1022 OpKernelContext* cc_ctx = reinterpret_cast<OpKernelContext*>(ctx); in set_tensor_data() local