Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 55 for tensors_ (0.19 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.h

        std::vector<TFE_TensorHandle*> result;
        result.reserve(tensors_.size());
        for (const TensorHandlePtr& tensor : tensors_) {
          result.emplace_back(tensor.get());
        }
        return result;
      }
    
     private:
      ParallelTensor(const ParallelDevice& device,
                     std::vector<TensorHandlePtr> tensors,
                     absl::Span<const int64_t> shape, const TF_DataType dtype)
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      for (int component_index = 0; component_index < tensors_.size();
           ++component_index) {
        // TODO(allenl): Add a C API for summarizing tensors. Currently custom
        // devices limiting themselves to a C API (for ABI compatibility) would need
        // to implement summarization for component tensors themselves.
        ImmediateExecutionTensorHandle* component =
            tensorflow::unwrap(tensors_[component_index].get());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients.cc

          });
    }
    bool Tape::ShouldRecord(
        absl::Span<const AbstractTensorHandle* const> tensors) const {
      std::vector<int64_t> tensor_ids(tensors.size());
      std::vector<tensorflow::DataType> tensor_dtypes(tensors.size());
      for (int i = 0; i < tensors.size(); i++) {
        tensor_ids[i] = ToId(tensors[i]);
        tensor_dtypes[i] = tensors[i]->DataType();
      }
      return GradientTape::ShouldRecord(tensor_ids, tensor_dtypes);
    }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/gradients.h

                           const string& op_name = "");
      // Returns whether any tensor in a list of tensors is being watched and has
      // a trainable dtype.
      bool ShouldRecord(
          absl::Span<const AbstractTensorHandle* const> tensors) const;
      // Unwatches this tensor on the tape. Mainly used for cleanup when deleting
      // eager tensors.
      void DeleteTrace(const AbstractTensorHandle*);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 6.9K bytes
    - Viewed (0)
  5. tensorflow/c/eager/parallel_device/parallel_device.h

    // an error.
    //
    // All component tensors must have the same dtype. Currently they must also have
    // the same shape, although this requirement may be relaxed in the future.
    //
    // `device_name` must not name an existing physical or custom device (see
    // the documentation for TFE_RegisterCustomDevice for more information).
    //
    // Tensors may be copied on or off the device explicitly using
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Jun 04 21:49:16 UTC 2020
    - 2.9K bytes
    - Viewed (0)
  6. tensorflow/c/eager/parallel_device/parallel_device.cc

                                           TFE_TensorHandle* tensor,
                                           TF_Status* status, void* device_info) {
      TF_SetStatus(
          status, TF_UNIMPLEMENTED,
          absl::StrCat("Trying to copy a tensor ",
                       tensorflow::unwrap(tensor)->DebugString(),
                       " on to a parallel device. Pack non-parallel "
                       "tensors for each device into a parallel tensor explicitly.")
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api.h

    // some input tensors are not on that device.
    // LINT.IfChange
    // Note: Keep in sync with internal copy of enum in eager/context.h.
    typedef enum TFE_ContextDevicePlacementPolicy {
      // Running operations with input tensors on the wrong device will fail.
      TFE_DEVICE_PLACEMENT_EXPLICIT = 0,
      // Copy the tensor to the right device but log a warning.
      TFE_DEVICE_PLACEMENT_WARN = 1,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  8. RELEASE.md

            `strategy.scope()`.
    *   `TensorRT`
        *   [TensorRT 6.0](https://developer.nvidia.com/tensorrt#tensorrt-whats-new)
            is now supported and enabled by default. This adds support for more
            TensorFlow ops including Conv3D, Conv3DBackpropInputV2, AvgPool3D,
            MaxPool3D, ResizeBilinear, and ResizeNearestNeighbor. In addition, the
            TensorFlow-TensorRT python conversion API is exported as
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Oct 22 14:33:53 UTC 2024
    - 735.3K bytes
    - Viewed (0)
  9. tensorflow/c/c_api.cc

        // Input tensors
        const std::vector<std::pair<string, Tensor>>& input_pairs,
        // Output tensors
        const std::vector<string>& output_tensor_names, TF_Tensor** c_outputs,
        // Target nodes
        const std::vector<string>& target_oper_names, TF_Buffer* run_metadata,
        TF_Status* status) {
      const int noutputs = output_tensor_names.size();
      std::vector<Tensor> outputs(noutputs);
      Status result;
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/immediate_execution_context.h

    // Note: Keep in sync with exported copy of enum in eager/c_api.h.
    enum ContextDevicePlacementPolicy {
      // Running operations with input tensors on the wrong device will fail.
      DEVICE_PLACEMENT_EXPLICIT = 0,
      // Copy the tensor to the right device but log a warning.
      DEVICE_PLACEMENT_WARN = 1,
      // Silently copy the tensor, which has a performance cost since the operation
      // will be blocked till the copy completes. This is the default policy.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.3K bytes
    - Viewed (0)
Back to top