- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 24 for Buffer (0.21 sec)
-
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/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/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) -
tensorflow/c/c_api_experimental.cc
// functions, and their send/recvs corountines). config.set_inter_op_parallelism_threads(1); TF_Buffer* ret = TF_NewBuffer(); TF_CHECK_OK(MessageToBuffer(config, ret)); return ret; } TF_Buffer* TF_CreateRunOptions(unsigned char enable_full_trace) { tensorflow::RunOptions options; if (enable_full_trace) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
TF_NewStatus(), TF_DeleteStatus); std::unique_ptr<TFE_ContextOptions, decltype(&TFE_DeleteContextOptions)> opts( TFE_NewContextOptions(), TFE_DeleteContextOptions); std::unique_ptr<TF_Buffer, decltype(&TF_DeleteBuffer)> config( TF_CreateConfig( /*xla*/ false, /* gpu_memory_allow_growth */ true, /* num_cpu_devices */ 2), TF_DeleteBuffer);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc
delete posix_file; } static int64_t Read(const TF_RandomAccessFile* file, uint64_t offset, size_t n, char* buffer, TF_Status* status) { auto posix_file = static_cast<PosixFile*>(file->plugin_file); char* dst = buffer; int64_t read = 0; while (n > 0) { // Some platforms, notably macs, throw `EINVAL` if `pread` is asked to read // more than fits in a 32-bit integer.
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/c_test_util.cc
TF_Status* s = TF_NewStatus(); TF_Buffer* buffer = TF_NewBuffer(); TF_GraphToGraphDef(graph, buffer, s); bool ret = TF_GetCode(s) == TF_OK; EXPECT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); if (ret) ret = graph_def->ParseFromArray(buffer->data, buffer->length); TF_DeleteBuffer(buffer); TF_DeleteStatus(s); return ret; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/experimental/grappler/grappler.cc
#include <unordered_set> #include <vector> #include "absl/status/status.h" #include "tensorflow/c/c_api_macros.h" #include "tensorflow/c/experimental/grappler/grappler_internal.h" #include "tensorflow/c/tf_buffer.h" #include "tensorflow/c/tf_buffer_internal.h" #include "tensorflow/c/tf_status.h" #include "tensorflow/c/tf_status_helper.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/framework/op.h"
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 15K bytes - Viewed (1) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
} PJRT_Buffer* TF_GetPjRtCBuffer(TF_Tensor* c_tensor, TF_Status* status) { tensorflow::Tensor tensor; auto s = tensorflow::TF_TensorToTensor(c_tensor, &tensor); if (!s.ok()) { status->status = s; return nullptr; } absl::StatusOr<PJRT_Buffer*> c_buffer = tensorflow::GetPjRtCBufferFromTensor(&tensor); if (!c_buffer.ok()) { status->status = c_buffer.status();
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/experimental/grappler/grappler_test.cc
#include "tensorflow/core/platform/test.h" namespace tensorflow { namespace grappler { namespace { void optimize_func(void* optimizer, const TF_Buffer* graph_buf, const TF_GrapplerItem* item, TF_Buffer* optimized_graph_buf, TF_Status* tf_status) {} void PopulateDefaultParam(TP_OptimizerRegistrationParams* params) {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Thu Apr 13 22:30:58 GMT 2023 - 11.6K bytes - Viewed (0)