Searched refs:TracingContext (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api.cc | 54 using tensorflow::tracing::TracingContext;
95 if (!isa<TracingContext>(ctx)) { in PYBIND11_MODULE()
102 return dyn_cast<TracingContext>(ctx); in PYBIND11_MODULE()
132 py::class_<TracingContext, AbstractContext>(m, "TracingContext") in PYBIND11_MODULE()
134 [](TracingContext* self, DataType dtype) { in PYBIND11_MODULE()
142 .def("Finalize", [](TracingContext* self, py::handle& outputs) { in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_unified_experimental.cc | 74 static TracingContext* CreateTracingExecutionContext(const char* fn_name, in CreateTracingExecutionContext()
107 using tensorflow::tracing::TracingContext;
124 TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(ctx)); in TF_FinalizeFunction()
141 TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(func)); in TF_AddFunctionParameter()
|
| D | c_api_unified_experimental_internal.h | 105 class TracingContext : public AbstractContext {
107 explicit TracingContext(AbstractContextKind kind) : AbstractContext(kind) {} in TracingContext() function
124 typedef TracingContext* (*FactoryFunction)(const char* fn_name, TF_Status*);
|
| D | unified_api_test.cc | 149 Status s = dyn_cast<tracing::TracingContext>(ctx.get())->AddParameter( in TEST_P()
176 s = dyn_cast<tracing::TracingContext>(ctx.get())->AddParameter( in TEST_P()
|
| D | unified_api_testutil.cc | 44 TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingContext>(ctx)->AddParameter( in CreateParamsForInputs()
88 TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingContext>(func_ctx.get()) in RunModel()
|
| D | c_api_unified_experimental_graph.cc | 346 class GraphContext : public TracingContext {
349 : TracingContext(kGraph), in GraphContext()
434 static TracingContext* GraphTracingFactory(const char* name, TF_Status* s) { in GraphTracingFactory()
|
| D | gradients_util.cc | 230 TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingContext>(ctx)->AddParameter( in CreateParamsForInputs()
274 TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingContext>(func_ctx.get()) in RunModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/ |
| D | c_api_unified_experimental_mlir_registration.cc | 18 using tensorflow::tracing::TracingContext;
21 TracingContext* MlirTracingFactory(const char* fn_name, TF_Status* s);
|
| D | c_api_unified_experimental_mlir.cc | 73 using tensorflow::tracing::TracingContext;
241 class MlirFunctionContext : public TracingContext {
244 : TracingContext(kMlir), in MlirFunctionContext()
679 TracingContext* MlirTracingFactory(const char* fn_name, TF_Status* s) { in MlirTracingFactory()
|