Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 46 for Tensor (0.07 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/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)
  3. tensorflow/c/eager/dlpack.cc

        status->status = tensorflow::errors::InvalidArgument(
            "DLPack doesn't support ", handle->TypeString(), " tensor");
        return nullptr;
      }
      const tensorflow::Tensor* tensor;
      status->status = handle->Tensor(&tensor);
      if (!status->status.ok()) {
        return nullptr;
      }
      return tensor;
    }
    
    // Deleter for DLManagedTensor
    void DLManagedTensorDeleter(DLManagedTensor* arg) {
      TfDlManagedTensorCtx* owner =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.9K bytes
    - Viewed (0)
  4. 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)
  5. tensorflow/c/eager/c_api_unified_experimental_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
      TFE_DeleteContextOptions(opts);
    
      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
    
      // Build an abstract input tensor.
      TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get());
      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
      TFE_TensorHandle* t = TestScalarTensorHandle(eager_ctx, 2.0f);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 39.1K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.h

                                                       TF_Status* status);
    
    // A handle to a tensor on a device.
    //
    // Like a TF_Tensor, a TFE_TensorHandle refers to a tensor with a value, shape,
    // type etc. Unlike a TF_Tensor, a TFE_TensorHandle may refer to such tensors
    // placed in the memory of different devices or remote address spaces.
    typedef struct TFE_TensorHandle TFE_TensorHandle;
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  7. tensorflow/c/c_api_experimental.cc

      if (VLOG_IS_ON(1) && status->status.ok()) {
        tensorflow::Tensor tensor;
        if (tensorflow::TF_TensorToTensor(ret, &tensor).ok()) {
          VLOG(1) << "Dequeued tensor content: " << tensor.DebugString();
        }
      }
      return ret;
    }
    
    void TF_EnqueueNamedTensor(TF_Session* session, int tensor_id,
                               TF_Tensor* tensor, TF_Status* status) {
      assert(session);
      {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  8. tensorflow/c/eager/immediate_execution_context.h

      // ops are implemented.
      //===--------------------------------------------------------------------===//
      // Create an abstract tensor handle from tensorflow::Tensor.
      virtual ImmediateExecutionTensorHandle* CreateLocalHandleFromTFTensor(
          tensorflow::Tensor& t, const char* d_name) = 0;
    
      // Convert a TFRT TensorHandle to tensorflow::TensorHandle.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api.cc

            "TFE_TensorHandleDeviceMemorySize may not be called on a ",
            handle->TypeString(), " tensor handle.");
        return 0;
      }
      const tensorflow::Tensor* tensor;
      status->status = handle->Tensor(&tensor);
      if (!status->status.ok()) {
        return 0;
      }
      return tensor->TotalBytes();
    }
    
    TFE_Op* TFE_NewOp(TFE_Context* ctx, const char* op_or_function_name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_experimental.h

                                                 TF_Status* status,
                                                 void* device_info);
    
      // Method to copy a tensor from the custom device to a target device.
      TFE_TensorHandle* (*copy_tensor_from_device)(TFE_Context* context,
                                                   TFE_TensorHandle* tensor,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 39.5K bytes
    - Viewed (0)
Back to top