Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 19 for handlers (0.16 sec)

  1. tensorflow/c/c_api.cc

      if (!status->status.ok()) {
        delete lib_handle;
        return nullptr;
      }
      return lib_handle;
    }
    
    TF_Buffer TF_GetOpList(TF_Library* lib_handle) { return lib_handle->op_list; }
    
    void TF_DeleteLibraryHandle(TF_Library* lib_handle) {
      if (lib_handle == nullptr) return;
      tensorflow::port::Free(const_cast<void*>(lib_handle->op_list.data));
      delete lib_handle;
    }
    
    TF_Buffer* TF_GetAllOpList() {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/dlpack_test.cc

      dltensor_in->strides = strides.data();
      TFE_TensorHandle* handle = TFE_HandleFromDLPack(&dlm_in, status, ctx);
      ASSERT_NE(handle, nullptr)
          << TF_Message(status) << " (shape=[" << absl::StrJoin(shape, ",")
          << "], strides=[" << absl::StrJoin(strides, ",") << "])";
    
      auto* dlm_out =
          static_cast<DLManagedTensor*>(TFE_HandleToDLPack(handle, status));
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Jun 30 03:04:46 GMT 2023
    - 4.4K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/gradients/nn_grad.cc

    #include "tensorflow/c/experimental/gradients/nn_grad.h"
    
    #include "absl/types/span.h"
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/eager/immediate_execution_context.h"
    #include "tensorflow/c/eager/immediate_execution_tensor_handle.h"
    #include "tensorflow/c/experimental/ops/array_ops.h"
    #include "tensorflow/c/experimental/ops/math_ops.h"
    #include "tensorflow/c/experimental/ops/nn_ops.h"
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/immediate_execution_tensor_handle.cc

    limitations under the License.
    ==============================================================================*/
    
    #include "tensorflow/c/eager/immediate_execution_tensor_handle.h"
    
    namespace tensorflow {
    
    std::string ImmediateExecutionTensorHandle::DebugString() const {
      PartialTensorShape shape;
      std::string shape_string;
      if (Shape(&shape).ok()) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 2.1K bytes
    - Viewed (0)
  5. tensorflow/c/eager/dlpack.cc

    };
    
    // Gets tensor from eager tensor handle.
    const Tensor* GetTensorFromHandle(TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
      }
      tensorflow::TensorHandle* handle =
          tensorflow::TensorHandleFromInterface(tensorflow::unwrap(h));
      if (handle->Type() != TensorHandle::LOCAL) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  6. tensorflow/c/eager/gradients.cc

    }
    
    TapeTensor::TapeTensor(AbstractTensorHandle* handle) : handle_(handle) {
      handle_->Ref();
    }
    TapeTensor::TapeTensor(const TapeTensor& other) {
      handle_ = other.handle_;
      handle_->Ref();
    }
    TapeTensor::~TapeTensor() { handle_->Unref(); }
    
    int64_t TapeTensor::GetID() const { return ToId(handle_); }
    
    tensorflow::DataType TapeTensor::GetDType() const {
      return handle_->DataType();
    }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/abstract_tensor_handle.cc

                          ", dtype=", DataType_Name(DataType()),
                          ", type=", FullType().DebugString(), ")");
    }
    
    Status AbstractTensorHandle::TensorHandleStatus() const {
      // Tensor handles in current runtime don't carry error info and this method
      // should always return OK status.
      return absl::OkStatus();
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 1.4K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_distributed_test.cc

      // Pack 3 variable handles into one TFE_TensorHandle.
      // When remote is false, function device is placed on task0. Handle types are
      // REMOTE, REMOTE, LOCAL on task0. When remote is true, function device is
      // placed on task1, Handle types are LOCAL, REMOTE, LOCAL on task1.
      int num_replicas = 3;
      std::vector<TFE_TensorHandle*> handles = {h0, h1, h2};
      TFE_TensorHandle* packed_handle =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  9. tensorflow/c/eager/gradient_checker.cc

    ==============================================================================*/
    #include "tensorflow/c/eager/gradient_checker.h"
    
    #include <memory>
    
    #include "absl/types/span.h"
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/experimental/ops/math_ops.h"
    #include "tensorflow/c/tf_tensor.h"
    
    namespace tensorflow {
    namespace gradients {
    
    using namespace std;
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/gradients_test.cc

    #include "tensorflow/c/eager/gradients.h"
    
    #include <memory>
    
    #include "absl/container/flat_hash_set.h"
    #include "absl/types/span.h"
    #include "tensorflow/c/eager/abstract_context.h"
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/eager/c_api_experimental.h"
    #include "tensorflow/c/eager/c_api_test_util.h"
    #include "tensorflow/c/eager/c_api_unified_experimental.h"
    #include "tensorflow/c/eager/c_api_unified_experimental_internal.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7K bytes
    - Viewed (0)
Back to top