Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for TracingContext (0.24 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
Back to top