- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for DataType (0.17 sec)
-
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/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/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) -
tensorflow/c/eager/immediate_execution_context.h
virtual AbstractTensorInterface* CreateBoolScalar(bool value) = 0; // Tensor creation functions virtual AbstractTensorInterface* CreateTensor( DataType dtype, absl::Span<const int64_t> dim_sizes) = 0; typedef void (*MemoryReleaser)(void* data, size_t len, void* arg); // Create a tensor instance from the given data buffer and description.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 06 08:34:00 GMT 2023 - 12.3K bytes - Viewed (0)