Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for space (0.14 sec)

  1. tensorflow/c/c_api_experimental.cc

      // Update the vector with information from `input_tensors` if provided.
      if (input_tensors != nullptr) {
        // Note that we take the address of the elements in `all_input_tensors`
        // below. Allocate enough space so that no reallocation happens, which will
        // make the pointers invalid.
        all_input_tensors.reserve(num_inputs);
        for (int i = 0; i < num_inputs; ++i) {
          if (input_tensors[i] == nullptr) continue;
    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/eager/gradient_checker.cc

      int num_elems = TF_TensorElementCount(theta_tensor);
      vector<float> theta_data(num_elems);
      memcpy(theta_data.data(), TF_TensorData(theta_tensor),
             TF_TensorByteSize(theta_tensor));
    
      // Initialize space for the numerical gradient.
      vector<float> dtheta_approx(num_elems);
    
      // Get theta shape and store in theta_dims.
      int num_dims = TF_NumDims(theta_tensor);
      vector<int64_t> theta_dims(num_dims);
    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)
  3. tensorflow/c/eager/gradients.cc

      return handle_->DataType();
    }
    AbstractTensorHandle* TapeTensor::GetHandle() const { return handle_; }
    
    AbstractTensorHandle* TapeTensor::ZerosLike() const { return nullptr; }
    
    class TapeVSpace
        : public eager::VSpace<AbstractTensorHandle, GradientFunction, TapeTensor> {
     public:
      explicit TapeVSpace(AbstractContext* ctx) : ctx_(ctx) {}
      ~TapeVSpace() override {}
    
      // Returns the number of elements in the gradient tensor.
    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)
  4. tensorflow/c/eager/c_api_experimental.cc

    }
    
    void TFE_HostAddressSpace(TFE_Context* ctx, TF_Buffer* buf) {
      auto address_space = tensorflow::DeviceNameUtils::AddressSpace(
          tensorflow::unwrap(ctx)->HostCPUParsedName());
      auto str = tensorflow::DeviceNameUtils::ParsedNameToString(address_space);
      void* data = tensorflow::port::Malloc(str.length());
      str.copy(static_cast<char*>(data), str.length(), 0);
      buf->data = data;
    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)
Back to top