- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 52 for buffer (0.4 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
#ifndef TENSORFLOW_C_EXPERIMENTAL_NEXT_PLUGGABLE_DEVICE_TENSOR_PJRT_BUFFER_UTIL_H_ #define TENSORFLOW_C_EXPERIMENTAL_NEXT_PLUGGABLE_DEVICE_TENSOR_PJRT_BUFFER_UTIL_H_ #include "xla/pjrt/c/pjrt_c_api.h" #include "xla/pjrt/pjrt_c_api_client.h" #include "tensorflow/core/framework/tensor.h" namespace tensorflow { StatusOr<PJRT_Buffer*> GetPjRtCBufferFromTensor(const Tensor* tensor);
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 09 19:12:59 GMT 2023 - 1.5K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
av_tensor->SetBuffer(std::move(buffer)); TF_ASSERT_OK_AND_ASSIGN(auto c_buffer, GetPjRtCBufferFromTensor(&tensor)); EXPECT_THAT(c_buffer, NotNull()); } TEST(TensorPjRtBufferUtilTest, SetPjRtCBufferToTensorNotAsyncValueTensor) { tensorflow::Tensor tensor(DT_FLOAT, {1}); TF_ASSERT_OK_AND_ASSIGN(auto pjrt_client, xla::GetCApiClient(DEVICE_CPU)); PJRT_Buffer* c_buffer = CreateCBuffer();
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/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
if (c_api_buffer == nullptr) { return absl::InternalError( "The PjRtBuffer in the tensor is not type PjRtCApiBuffer."); } return c_api_buffer->c_buffer(); } absl::Status SetPjRtCBufferToTensor(PJRT_Buffer* c_buffer, xla::PjRtCApiClient* c_api_client, Tensor* tensor) { auto buffer = std::make_unique<xla::PjRtCApiBuffer>(c_api_client, c_buffer);
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 3.7K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc
} TEST(RamFileBlockCacheTest, ValidateAndUpdateFileSignature) { int calls = 0; auto fetcher = [&calls](const string& filename, size_t offset, size_t n, char* buffer, TF_Status* status) -> int64_t { calls++; memset(buffer, 'x', n); TF_SetStatus(status, TF_OK, ""); return n; }; string filename = "file"; tf_gcs_filesystem::RamFileBlockCache cache(16, 32, 0, fetcher); std::vector<char> out;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:57 GMT 2021 - 23.2K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/posix/copy_file_portable.cc
// Transfers up to `size` bytes from `dst_fd` to `src_fd`. // // This method uses a temporary buffer to hold contents. int CopyFileContents(int dst_fd, int src_fd, off_t size) { // Use a copy buffer of 128KB but don't store it on the stack. constexpr static size_t kPosixCopyFileBufferSize = 128 * 1024; std::unique_ptr<char[]> buffer(new char[kPosixCopyFileBufferSize]); off_t offset = 0; int bytes_transferred = 0; int rc = 1;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Nov 22 21:23:55 GMT 2019 - 1.9K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
graph_def->length)); TF_Buffer graph_def_buffer; graph_def_buffer.data = reinterpret_cast<const void*>(&graph_def_proto); graph_def_buffer.length = sizeof(tensorflow::GraphDef*); TF_ImportGraphDefResults* results = TF_GraphImportGraphDefWithResultsNoSerialization(graph, &graph_def_buffer, opts, s);
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/c_op_requires.h
#define TF_CLEANUP_AND_RETURN_IF_ERROR(C_STATUS, BUFFER, __VA_ARGS__) \ do { \ ::tensorflow::Status _s(__VA_ARGS__); \ if (TF_PREDICT_TRUE(!_s.ok())) { \ TF_DeleteStatus(C_STATUS); \ TF_DeleteBuffer(BUFFER); \
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Aug 02 21:35:06 GMT 2022 - 2.3K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler_internal.h
}; using OwnedTFStatus = std::unique_ptr<TF_Status, TFStatusDeleter>; struct TFBufferDeleter { void operator()(TF_Buffer* buf) const { TF_DeleteBuffer(buf); } }; using OwnedTFBuffer = std::unique_ptr<TF_Buffer, TFBufferDeleter>; class CGraphOptimizer : public CustomGraphOptimizer { public: explicit CGraphOptimizer(TP_Optimizer optimizer, const char* device_type)
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Jun 08 08:58:23 GMT 2022 - 3.5K bytes - Viewed (1) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache.cc
size_t n, char* buffer, TF_Status* status) { if (n == 0) { TF_SetStatus(status, TF_OK, ""); return 0; } if (!IsCacheEnabled() || (n > max_bytes_)) { // The cache is effectively disabled, so we pass the read through to the // fetcher without breaking it up into blocks. return block_fetcher_(filename, offset, n, buffer, status); }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 16 01:39:09 GMT 2020 - 11.1K bytes - Viewed (0) -
tensorflow/c/c_api.cc
} *result = TF_TensorFromTensor(**status_or, &status->status); return status->status.ok(); } TF_ApiDefMap* TF_NewApiDefMap(TF_Buffer* op_list_buffer, TF_Status* status) { tensorflow::OpList op_list; if (!op_list.ParseFromArray(op_list_buffer->data, op_list_buffer->length)) { status->status = InvalidArgument("Unparseable OpList"); return nullptr; } status->status = absl::OkStatus();
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)