- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for uint32 (0.18 sec)
-
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; case TF_DataType::TF_BFLOAT16:
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/dlpack_test.cc
std::vector<int64_t> shape, std::vector<int64_t> strides) { size_t num_elements = 1; for (int i = 0; i < static_cast<int32_t>(shape.size()); ++i) { num_elements *= shape[i]; } std::vector<float> data(num_elements); for (size_t j = 0; j < num_elements; ++j) { data[j] = j; } DLManagedTensor dlm_in = {};
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/c_api.cc
TF_DEVICELIST_METHOD(const char*, TF_DeviceListType, device_type().c_str(), nullptr); TF_DEVICELIST_METHOD(int64_t, TF_DeviceListMemoryBytes, memory_limit(), -1); TF_DEVICELIST_METHOD(uint64_t, TF_DeviceListIncarnation, incarnation(), 0); #undef TF_DEVICELIST_METHOD } // end extern "C" // -------------------------------------------------------------------------- // New Graph and Session API
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc
// more than fits in a 32-bit integer. size_t requested_read_length; if (n > INT32_MAX) requested_read_length = INT32_MAX; else requested_read_length = n; // `pread` returns a `ssize_t` on POSIX, but due to interface being // cross-platform, return type of `Read` is `int64_t`. int64_t r = int64_t{pread(posix_file->fd, dst, requested_read_length,
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/eager/gradient_checker.cc
namespace gradients { using namespace std; // ================== Helper functions ================= // Fills data with values [start,end) with given step size. void Range(vector<int32_t>* data, int32_t start, int32_t end, int32_t step = 1) { for (int32_t i = start; i < end; i += step) { (*data)[i] = i; } } // Fills out_dims with the dimensions of the given tensor. void GetDims(const TF_Tensor* t, int64_t* out_dims) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
} auto pjrt_client = xla::GetCApiClient(DEVICE_CPU); CHECK_OK(pjrt_client.status()); auto c_api_client = down_cast<xla::PjRtCApiClient*>(pjrt_client->get()); std::vector<int32_t> data(1, 0); xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1}); auto buffer = c_api_client->pjrt_c_client()->client->BufferFromHostBuffer( data.data(), shape.element_type(), shape.dimensions(),
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
} std::unique_ptr<ParallelTensor> ParallelDevice::DeviceIDs( TFE_Context* context, TF_Status* status) const { std::vector<int32_t> ids; ids.reserve(num_underlying_devices()); for (int i = 0; i < num_underlying_devices(); ++i) { ids.push_back(i); } return ScalarsFromSequence<int32_t>(ids, context, status); } absl::optional<std::vector<std::unique_ptr<ParallelTensor>>>
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/c_api_test_util.cc
tensorflow::uint64 send_device_incarnation) { TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, op_name.c_str(), status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TF_DeleteStatus(status); TFE_OpSetAttrType(op, "tensor_type", TF_INT32); TFE_OpSetAttrString(op, "tensor_name", "dummy", 5);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/c_api_function_test.cc
ASSERT_TRUE(out != nullptr); EXPECT_EQ(TF_INT32, TF_TensorType(out)); EXPECT_EQ(0, TF_NumDims(out)); // scalar ASSERT_EQ(sizeof(int32_t), TF_TensorByteSize(out)); int32_t* output_contents = static_cast<int32_t*>(TF_TensorData(out)); EXPECT_EQ(expected_results[i], *output_contents); } } void CompareInt32Tensor(const std::vector<int32_t>& expected, TF_Tensor* t) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 20 22:08:54 GMT 2023 - 63.6K bytes - Viewed (6) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
namespace tf_random_access_file { using ReadFn = std::function<int64_t(const std::string& path, uint64_t offset, size_t n, char* buffer, TF_Status* status)>; typedef struct GCSFile { const std::string path; const bool is_cache_enable; const uint64_t buffer_size; ReadFn read_fn; absl::Mutex buffer_mutex; uint64_t buffer_start ABSL_GUARDED_BY(buffer_mutex);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 23 06:55:53 GMT 2023 - 46.9K bytes - Viewed (0)