- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for DataType (0.2 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/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/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) -
tensorflow/c/eager/abstract_tensor_handle.h
explicit AbstractTensorHandle(AbstractTensorHandleKind kind) : kind_(kind) {} ~AbstractTensorHandle() override {} public: // Returns tensor dtype. virtual tensorflow::DataType DataType() const = 0; // Returns the status of the tensor handle. If it is a tfrt::TensorHandle, // the tensor handle can be an error and return non-OK status. virtual tensorflow::Status TensorHandleStatus() const;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Mar 03 00:30:36 GMT 2023 - 3K bytes - Viewed (0) -
tensorflow/c/eager/gradients_internal.h
ForwardOperation*); Status SetAttrBool(AbstractOperation*, const char* attr_name, bool value, ForwardOperation*); Status SetAttrType(AbstractOperation*, const char* attr_name, DataType value, ForwardOperation*); Status SetAttrShape(AbstractOperation*, const char* attr_name, const int64_t* dims, const int num_dims, ForwardOperation*);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:35 GMT 2021 - 4.2K bytes - Viewed (0) -
tensorflow/c/c_api_macros.h
#else #define TF_CAPI_EXPORT __attribute__((visibility("default"))) #endif // TF_CAPI_WEAK #endif // _WIN32 #endif // SWIG // TF_Bool is the C API typedef for unsigned char, while TF_BOOL is // the datatype for boolean tensors. #ifndef TF_Bool #define TF_Bool unsigned char #endif // TF_Bool // Macro used to calculate struct size for maintaining ABI stability across // different struct implementations.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sat May 13 04:44:45 GMT 2023 - 1.6K bytes - Viewed (1) -
tensorflow/c/eager/gradients.h
class TapeTensor { public: explicit TapeTensor(AbstractTensorHandle* handle); TapeTensor(const TapeTensor& other); ~TapeTensor(); int64_t GetID() const; tensorflow::DataType GetDType() const; AbstractTensorHandle* ZerosLike() const; AbstractTensorHandle* GetHandle() const; private: AbstractTensorHandle* handle_; };
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 10:27:05 GMT 2022 - 6.9K bytes - Viewed (0) -
tensorflow/c/c_api_internal.h
const int64_t** shapes, const int* ranks, const TF_DataType* types, TF_Status* status); void RecordMutation(TF_Graph* graph, const TF_Operation& op, const char* mutation_type)
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sat May 13 00:49:12 GMT 2023 - 7.6K bytes - Viewed (0)