- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for TracingContext (0.24 sec)
-
tensorflow/c/eager/c_api_unified_experimental.cc
TF_OutputList* outputs, TF_Status* s) { AbstractFunction* func; TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(ctx)); if (!tracing_ctx) { tsl::Set_TF_Status_from_Status( s, tensorflow::errors::InvalidArgument( "Only TracingContext can be converted into a function.")); return nullptr; } tsl::Set_TF_Status_from_Status(s,
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_internal.h
// This holds the context for the execution: dispatching operations either to an // MLIR implementation or to a graph implementation. class TracingContext : public AbstractContext { protected: explicit TracingContext(AbstractContextKind kind) : AbstractContext(kind) {} public: // Add a function parameter and return the corresponding tensor.
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Nov 13 22:20:40 GMT 2020 - 5.2K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.h
#include "tensorflow/c/tf_status_helper.h" #include "tensorflow/c/tf_tensor.h" #include "tensorflow/core/platform/status.h" namespace tensorflow { // Builds and returns a `TracingContext` using the default tracing impl. AbstractContext* BuildFunction(const char* fn_name); // Creates parameters (placeholders) in the tracing `ctx` using the shape and // dtype of `inputs`.
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_test.cc
} AbstractContextPtr ctx(BuildFunction("test_fn")); AbstractTensorHandlePtr x; { tracing::TracingTensorHandle* x_raw = nullptr; PartialTensorShape shape; Status s = dyn_cast<tracing::TracingContext>(ctx.get())->AddParameter( DT_FLOAT, shape, &x_raw); ASSERT_EQ(errors::OK, s.code()) << s.message(); x.reset(x_raw); } PartialTensorShape shape; Status s = x->Shape(&shape);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
tracing::TracingTensorHandle* handle = nullptr; for (auto input : inputs) { PartialTensorShape shape; TF_RETURN_IF_ERROR(input->Shape(&shape)); TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingContext>(ctx)->AddParameter( input->DataType(), shape, &handle)); params->emplace_back(handle); } return absl::OkStatus(); } // Runs `model` maybe wrapped in a function.
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
}; // GraphContext wraps a TF_Graph modeling a single function and manages the // "execution" of operation, i.e. adding them to the function. class GraphContext : public TracingContext { public: explicit GraphContext(const char* name) : TracingContext(kGraph), graph_(new TF_Graph(), TF_DeleteGraph), name_(name) {} void Release() override { delete this; }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1)