- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for DataType (0.18 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.h
"Number of values did not match number of underlying devices."); return nullptr; } TF_DataType datatype_enum( static_cast<TF_DataType>(DataTypeToEnum<DataType>().value)); for (int device_index = 0; device_index < num_underlying_devices(); ++device_index) { auto device_value = absl::make_unique<DataType>(); *device_value = values[device_index];
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.h
#ifndef TENSORFLOW_C_EAGER_C_API_UNIFIED_EXPERIMENTAL_H_ #define TENSORFLOW_C_EAGER_C_API_UNIFIED_EXPERIMENTAL_H_ #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/tf_datatype.h" #include "tensorflow/c/tf_status.h" #ifdef __cplusplus extern "C" { #endif // =============================================================================
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:00 GMT 2021 - 7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.h
// Return a tensor handle with given type, values and dimensions. template <class T, TF_DataType datatype> TFE_TensorHandle* TestTensorHandleWithDims(TFE_Context* ctx, const T* data, const int64_t* dims, int num_dims) { TF_Status* status = TF_NewStatus(); TF_Tensor* t = TFE_AllocateHostTensor(ctx, datatype, dims, num_dims, status); memcpy(TF_TensorData(t), data, TF_TensorByteSize(t));
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Jul 17 23:43:59 GMT 2023 - 7.7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.h
// separate static method which returns a status. Variable(TFE_TensorHandle* handle, TF_DataType type) : handle_(handle), type_(type) {} // Helper for constructing a resource handle and wrapping it in a `Variable` // object. static Variable* Create(TFE_Context* context, TF_DataType type, const int64_t* dims, const int num_dims,
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
TF_CheckpointReader* reader, int index); // Get the number of variable in the checkpoint TF_CAPI_EXPORT extern int TF_CheckpointReaderSize(TF_CheckpointReader* reader); // Get the DataType of a variable TF_CAPI_EXPORT extern TF_DataType TF_CheckpointReaderGetVariableDataType( TF_CheckpointReader* reader, const char* name); // Read the shape of a variable and write to `dims`
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.h
Status SetAttrFloat(const char* attr_name, float value) override; Status SetAttrBool(const char* attr_name, bool value) override; Status SetAttrType(const char* attr_name, DataType value) override; Status SetAttrShape(const char* attr_name, const int64_t* dims, const int num_dims) override; Status SetAttrFunction(const char* attr_name,
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 3.7K bytes - Viewed (1) -
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 23 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 23 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0) -
tensorflow/c/c_api.h
const char* attr_name, TF_DataType value); TF_CAPI_EXPORT extern void TF_SetAttrTypeList(TF_OperationDescription* desc, const char* attr_name, const TF_DataType* values, int num_values);
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3) -
tensorflow/c/eager/unified_api_testutil.h
TFE_TensorHandle* input_eager = TestTensorHandleWithDims<T, datatype>(eager_ctx, data, dims, num_dims); *tensor = unwrap(TF_CreateAbstractTensorFromEagerTensor(input_eager, status.get())); return absl::OkStatus(); } // Return a scalar tensor handle with given value. template <class T, TF_DataType datatype> Status TestScalarTensorHandle(AbstractContext* ctx, const T value,
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0)