- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for DataType (0.19 sec)
-
tensorflow/c/eager/dlpack.cc
break; case TF_DataType::TF_INT8: case TF_DataType::TF_INT16: case TF_DataType::TF_INT32: case TF_DataType::TF_INT64: dtype.code = DLDataTypeCode::kDLInt; break; case TF_DataType::TF_UINT8: case TF_DataType::TF_UINT16: case TF_DataType::TF_UINT32: case TF_DataType::TF_UINT64: dtype.code = DLDataTypeCode::kDLUInt; break;
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) -
tensorflow/c/eager/gradients.cc
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(); } AbstractTensorHandle* TapeTensor::GetHandle() const { return handle_; } AbstractTensorHandle* TapeTensor::ZerosLike() const { return nullptr; } class TapeVSpace
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) -
tensorflow/c/BUILD
], }), ) cc_library( name = "tf_datatype_hdrs", hdrs = ["tf_datatype.h"], visibility = ["//tensorflow:internal"], deps = [ ":c_api_macros_hdrs", ], ) cc_library( name = "tf_datatype", srcs = ["tf_datatype.cc"], hdrs = ["tf_datatype.h"], copts = tf_copts(), visibility = ["//visibility:public"], deps = [
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 27 18:00:18 GMT 2024 - 30.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
TF_DataType value) { auto iter = builder->attr_names.insert(attr_name).first; builder->Set(*iter, static_cast<tensorflow::DataType>(value)); } void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder, const char* attr_name, const TF_DataType* values, int num_values) { auto iter = builder->attr_names.insert(attr_name).first;
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) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
public: explicit GraphTensor(TF_Output output, TF_Graph* graph) : TracingTensorHandle(kGraph), output_(output), graph_(graph) {} tensorflow::DataType DataType() const override { return static_cast<tensorflow::DataType>(TF_OperationOutputType(output_)); } tensorflow::Status Shape( tensorflow::PartialTensorShape* shape) const override { DCHECK(shape != nullptr); TF_Status status;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/eager/c_api.cc
"TFE_DeleteTensorHandle", tensorflow::profiler::TraceMeLevel::kInfo); if (h) { tensorflow::unwrap(h)->Unref(); } } TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) { return static_cast<TF_DataType>(tensorflow::unwrap(h)->DataType()); } int TFE_TensorHandleNumDims(TFE_TensorHandle* h, TF_Status* status) { if (h == nullptr) {
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) -
tensorflow/c/eager/tape.h
} // Returns whether any tensor in a list of tensors is being watched and has // a trainable dtype. bool ShouldRecord(absl::Span<const int64_t> tensor_ids, absl::Span<const tensorflow::DataType> dtypes) const; // Adds this tensor to the list of watched tensors. // // This is a no-op if the tensor is already being watched either from an
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 02 12:40:29 GMT 2024 - 47.2K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental.h
// // Custom devices must use TFE_NewCustomDeviceTensorHandle instead. TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandleFromDeviceMemory( TFE_Context* ctx, const char* device_name, TF_DataType, const int64_t* dims, int num_dims, void* data, size_t len, void (*deallocator)(void* data, size_t len, void* arg), void* deallocator_arg, TF_Status* status);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
buf->data_deallocator = [](void* data, size_t length) { tensorflow::port::Free(data); }; status->status = absl::OkStatus(); } TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx, TF_DataType dtype, const int64_t* dims, int num_dims, TF_Status* status) { std::vector<int64_t> dimvec(num_dims); for (int i = 0; i < num_dims; ++i) {
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) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
return nullptr; } TFE_TensorHandleGetStatus(components[0].get(), status); if (!status->status.ok()) { return nullptr; } TF_DataType dtype = TFE_TensorHandleDataType(components[0].get()); // Verify that the TensorHandle's shape and dtype match all of the component // shapes and dtypes. for (TensorHandlePtr& component : components) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1)