- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 18 for DataType (0.2 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 30 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_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 30 12:39:09 GMT 2024 - Last Modified: Mon Jul 17 23:43:59 GMT 2023 - 7.7K 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 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:00 GMT 2021 - 7K 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 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0) -
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/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 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3) -
tensorflow/c/eager/abstract_operation.h
virtual Status SetAttrFloat(const char* attr_name, float value) = 0; virtual Status SetAttrBool(const char* attr_name, bool value) = 0; virtual Status SetAttrType(const char* attr_name, DataType value) = 0; virtual Status SetAttrShape(const char* attr_name, const int64_t* dims, const int num_dims) = 0; virtual Status SetAttrShape(const char* attr_name,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Jul 14 16:20:41 GMT 2021 - 6.8K bytes - Viewed (0) -
tensorflow/c/c_test.c
#include "tensorflow/c/kernels.h" // A create function. This will never actually get called in this test, it's // just nice to know that it compiles. void* create(TF_OpKernelConstruction* ctx) { TF_DataType type; TF_Status* s = TF_NewStatus(); TF_OpKernelConstruction_GetAttrType(ctx, "foobar", &type, s); TF_DeleteStatus(s); return NULL; }
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Apr 24 20:50:35 GMT 2024 - 2.8K bytes - Viewed (0) -
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 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0)