- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 45 for DataType (0.15 sec)
-
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/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/abstract_op_attrs.h
virtual bool GetBool(absl::string_view attr_name, bool* result) const = 0; virtual bool GetType(absl::string_view attr_name, DataType* result) const = 0; virtual Status GetTypeList( absl::string_view attr_name, absl::InlinedVector<DataType, 4>* type_list) const = 0; private: const AbstractOpAttrsKind kind_; }; } // namespace tensorflow
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed May 26 22:20:27 GMT 2021 - 2K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
} Status TapeOperation::SetAttrTypeList(const char* attr_name, const DataType* values, int num_values) { forward_op_.attrs.Set(attr_name, gtl::ArraySlice<const DataType>(values, num_values)); return parent_op_->SetAttrTypeList(attr_name, values, num_values); } Status TapeOperation::SetAttrBoolList(const char* attr_name,
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/eager/c_api_unified_experimental.cc
} void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name, TF_DataType value, TF_Status* s) { Status status = unwrap(op)->SetAttrType(attr_name, static_cast<DataType>(value)); TF_SetStatus(s, static_cast<TF_Code>(status.code()), tsl::NullTerminatedMessage(status)); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/checkpoint_reader.cc
TF_Status* out_status) const { Status status; if (reader_ != nullptr) { status = reader_->GetTensor(name, out_tensor); } else { tensorflow::DataType dtype; tensorflow::TensorShape shape; status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape); if (status.ok()) { out_tensor->reset(new Tensor(dtype, shape));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Aug 25 21:29:12 GMT 2023 - 5.5K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
EXPECT_EQ(TF_FLOAT, static_cast<TF_DataType>(tensorflow::DT_FLOAT)); EXPECT_EQ(TF_DOUBLE, static_cast<TF_DataType>(tensorflow::DT_DOUBLE)); EXPECT_EQ(TF_INT32, static_cast<TF_DataType>(tensorflow::DT_INT32)); EXPECT_EQ(TF_UINT8, static_cast<TF_DataType>(tensorflow::DT_UINT8)); EXPECT_EQ(TF_INT16, static_cast<TF_DataType>(tensorflow::DT_INT16)); EXPECT_EQ(TF_INT8, static_cast<TF_DataType>(tensorflow::DT_INT8));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
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/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/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)