Searched defs:cc_ctx (Results 1 – 5 of 5) sorted by relevance
106 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in GetAttrValue() local264 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_NumInputs() local269 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_NumOutputs() local275 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetInput() local290 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_InputRange() local300 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_SetOutput() local316 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetMutableOutput() local334 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetSerializedFunctionDefLibrary() local345 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetSerializedConfigProto() local359 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in TF_OpKernelConstruction_Failure() local[all …]
152 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in GetTrainingVariableMutex() local176 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_AssignVariable() local225 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_AssignRefVariable() local229 const ::tensorflow::Tensor& rhs) { in TF_AssignRefVariable()293 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_MaybeLockVariableInputMutexesInOrder() local354 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetInputTensorFromVariable() local382 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_OpKernelContext_ForwardRefInputToRefOutput() local400 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_GetInputByName() local420 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); in TF_OpKernelConstruction_GetAttrTensorShape() local439 auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); in TF_IsRefInput() 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
1405 OpKernelContext* cc_ctx = reinterpret_cast<OpKernelContext*>(ctx); in set_tensor_data() local
159 const Variant& b, Variant* out) { in Compute()