Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for TFE_DeleteOp (0.07 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

          "/job:localhost/replica:0/task:0/device:CPU:1"};
      ParallelDevice parallel_device(std::move(devices));
      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op(
          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_test.cc

            << TF_Message(status.get());
      }
    
      {
        // Try to pass a non-parallel tensor to TPUReplicatedOutput
        std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op(
            TFE_NewOp(context.get(), "TPUReplicatedOutput", status.get()),
            TFE_DeleteOp);
        if (TF_GetCode(status.get()) != TF_OK) return;
        TFE_OpSetAttrInt(op.get(), "num_replicas", 1);
        TFE_OpAddInput(op.get(), value_one.get(), status.get());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_cluster_test.cc

      CheckTFE_TensorHandleHasFloats(retval_task0, {7, 10, 15, 22});
    
      TFE_DeleteTensorHandle(retval_task0);
      TFE_DeleteTensorHandle(h0_task0);
      TFE_DeleteTensorHandle(retvals[0]);
    
      TFE_DeleteOp(matmul);
    
      TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx);
      TFE_ExecutorWaitForAllPendingNodes(executor, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_distributed_test.cc

      TFE_TensorHandle* retvals[1] = {nullptr};
      int num_retvals = 1;
      TFE_Execute(func, &retvals[0], &num_retvals, status);
      EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
      ASSERT_EQ(1, num_retvals);
      TFE_DeleteOp(func);
      TFE_DeleteTensorHandle(packed_handle);
      TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status);
      ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
      TFE_DeleteTensorHandle(retvals[0]);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 23.4K bytes
    - Viewed (0)
  5. tensorflow/c/eager/parallel_device/parallel_device_lib.h

        // on to one TFE_Op per device and just call TFE_ResetOp to avoid parsing
        // device names repeatedly.
        std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> const_op(
            TFE_NewOp(context, "Const", status), TFE_DeleteOp);
        if (TF_GetCode(status) != TF_OK) return nullptr;
        TFE_OpSetDevice(const_op.get(), underlying_devices_[device_index].c_str(),
                        status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.cc

      status->status = new_op->Reset(op_or_function_name, nullptr);
      if (!status->status.ok()) {
        new_op->Release();
        new_op = nullptr;
      }
      return tensorflow::wrap(new_op);
    }
    
    void TFE_DeleteOp(TFE_Op* op) {
      if (op == nullptr) {
        return;
      }
    
      tensorflow::unwrap(op)->Release();
    }
    
    const char* TFE_OpGetName(const TFE_Op* op, TF_Status* status) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  7. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

    #include "tsl/platform/errors.h"
    #include "tsl/platform/thread_annotations.h"
    
    namespace tensorflow {
    namespace parallel_device {
    namespace {
    
    class OpDeleter {
     public:
      void operator()(TFE_Op* to_delete) const { TFE_DeleteOp(to_delete); }
    };
    
    using OpPtr = std::unique_ptr<TFE_Op, OpDeleter>;
    
    class StatusDeleter {
     public:
      void operator()(TF_Status* to_delete) const { TF_DeleteStatus(to_delete); }
    };
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  8. tensorflow/c/eager/parallel_device/parallel_device.cc

    #include "tensorflow/c/tf_status.h"
    #include "tensorflow/c/tf_status_helper.h"
    
    namespace tensorflow {
    namespace parallel_device {
    namespace {
    
    class OpDeleter {
     public:
      void operator()(TFE_Op* to_delete) const { TFE_DeleteOp(to_delete); }
    };
    
    using OpPtr = std::unique_ptr<TFE_Op, OpDeleter>;
    
    using MaybeParallelTensorOwned =
        absl::variant<std::unique_ptr<ParallelTensor>, TensorHandlePtr>;
    
    using MaybeParallelTensorUnowned =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
Back to top