Searched refs:tracing_ctx (Results 1 – 1 of 1) sorted by relevance
124 TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(ctx)); in TF_FinalizeFunction() local125 if (!tracing_ctx) { in TF_FinalizeFunction()131 Set_TF_Status_from_Status(s, tracing_ctx->Finalize(unwrap(outputs), &func)); in TF_FinalizeFunction()141 TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(func)); in TF_AddFunctionParameter() local142 if (!tracing_ctx) { in TF_AddFunctionParameter()160 s, tracing_ctx->AddParameter(static_cast<DataType>(dtype), partial_shape, in TF_AddFunctionParameter()