Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 11 for bundle (0.19 sec)

  1. tensorflow/c/c_api_experimental.cc

        if (it != loaded_libs->end()) {
          lib_handle->lib_handle = it->second;
        } else {
          status->status =
              env->LoadDynamicLibrary(library_filename, &lib_handle->lib_handle);
          if (status->status.ok()) {
            TF_CHECK_OK(
                tensorflow::RegisterPluggableDevicePlugin(lib_handle->lib_handle));
          } else {
            delete lib_handle;
            return nullptr;
          }
        }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc

    // SECTION 2. Implementation for `TF_WritableFile`
    // ----------------------------------------------------------------------------
    namespace tf_writable_file {
    
    typedef struct PosixFile {
      const char* filename;
      FILE* handle;
    } PosixFile;
    
    static void Cleanup(TF_WritableFile* file) {
      auto posix_file = static_cast<PosixFile*>(file->plugin_file);
      plugin_memory_free(const_cast<char*>(posix_file->filename));
      delete posix_file;
    }
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Mar 24 20:08:23 GMT 2024
    - 15.8K bytes
    - Viewed (0)
  3. 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)
  4. 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)
  5. tensorflow/c/eager/c_api_distributed_test.cc

      EXPECT_EQ(TFE_TensorHandleDataType(packed_handle), TF_RESOURCE);
      EXPECT_EQ(TFE_TensorHandleNumDims(packed_handle, status), 0);
      EXPECT_EQ(TFE_TensorHandleNumElements(packed_handle, status), 1);
    
      const string composite_device_name =
          "/job:localhost/replica:0/task:0/device:COMPOSITE:0";
      EXPECT_EQ(TFE_TensorHandleDeviceName(packed_handle, status),
                composite_device_name);
    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)
  6. tensorflow/c/eager/c_api.cc

            ->DevicePointer();
      }
      // TODO(b/175427838): It would be nice to be able to use tensorflow::isa here.
      if (!tensorflow::TensorHandle::classof(unwrapped_handle)) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
      }
      tensorflow::TensorHandle* handle =
          tensorflow::TensorHandleFromInterface(unwrapped_handle);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  7. tensorflow/c/eager/c_api_test_util.cc

      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_TensorHandle* var_handle = nullptr;
      int num_retvals = 1;
      TFE_Execute(op, &var_handle, &num_retvals, status);
      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_DeleteOp(op);
      if (TF_GetCode(status) != TF_OK) return nullptr;
      CHECK_EQ(1, num_retvals);
      TF_DeleteStatus(status);
      return var_handle;
    }
    
    TFE_TensorHandle* CreateVariable(TFE_Context* ctx, float value,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  8. tensorflow/c/eager/c_api_experimental.cc

        tensorflow::ImmediateExecutionTensorHandle* unwrapped_handle =
            tensorflow::unwrap(handles[i]);
        if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) {
          // One of the inputs we're trying to pack is on a custom device. We'll let
          // the first custom device we see handle all of the packing.
          auto* custom_device_handle =
              tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
  9. tensorflow/c/experimental/gradients/math_grad.cc

    limitations under the License.
    ==============================================================================*/
    #include "tensorflow/c/experimental/gradients/math_grad.h"
    
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/eager/gradients.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"
    
    using std::vector;
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 15.2K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_experimental_test.cc

        // Read the value of tensor handle `handle_1`.
        float value = 0.0f;
        TF_Tensor* t = TFE_TensorHandleResolve(handle_1, status);
        ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        ASSERT_EQ(sizeof(float), TF_TensorByteSize(t));
        memcpy(&value, TF_TensorData(t), sizeof(float));
        TF_DeleteTensor(t);
        EXPECT_EQ(1.2f, value);
        TFE_DeleteTensorHandle(handle_1);
        TF_DeleteStatus(status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 03:14:26 GMT 2023
    - 31.5K bytes
    - Viewed (1)
Back to top