- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 104 for tensor (0.1 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
tensorflow/c/eager/c_api_debug.cc
TFE_TensorHandle* h, TF_Status* status) { tensorflow::TensorHandle* handle = TensorHandleFromInterface(tensorflow::unwrap(h)); const tensorflow::Tensor* tensor; status->status = handle->Tensor(&tensor); if (!status->status.ok()) { return nullptr; } std::vector<int64_t> dev_dims = TensorShapeAsVector(*handle, &status->status); if (!status->status.ok()) { return nullptr;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 2.5K bytes - Viewed (0) -
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) -
tensorflow/c/eager/immediate_execution_tensor_handle.h
// Returns a tensor for the handle. If tensor is remote, it will be copied. virtual AbstractTensorInterface* Resolve(absl::Status* status) = 0; std::string DebugString() const override; // Returns a Boolean hint indicating whether callers should prefer // `SummarizeValue` to resolving this handle and formatting the tensor. // // For example some tensor handles may represent distributed values, in which
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 4.3K bytes - Viewed (0)