- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 26 for buffer (0.2 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.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_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/eager/c_api_test.cc
const int BUFFER_SIZE = 10; char buffer[BUFFER_SIZE]; std::strncpy(buffer, "VALID", BUFFER_SIZE); TFE_OpSetAttrString(op, "padding", buffer, std::strlen(buffer)); // Overwriting value in "buffer", should be fine since TFE_Op // shouldn't be holding on to it. std::strncpy(buffer, "NHWC", BUFFER_SIZE); TFE_OpSetAttrString(op, "data_format", buffer, std::strlen(buffer));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/experimental/filesystem/modular_filesystem.cc
option_value.type_tag = TF_Filesystem_Option_Type_Buffer; option_value.num_values = values.size(); std::vector<TF_Filesystem_Option_Value_Union> option_values(values.size()); for (size_t i = 0; i < values.size(); i++) { memset(&option_values[i], 0, sizeof(option_values[i])); option_values[i].buffer_val.buf = const_cast<char*>(values[i].c_str()); option_values[i].buffer_val.buf_length = values[i].size(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 23.1K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
copy_size = (std::min)(n, static_cast<size_t>(buffer_end - offset)); memcpy(buffer, gcs_file->buffer.data() + (offset - gcs_file->buffer_start), copy_size); } bool consumed_buffer_to_eof = offset + copy_size >= buffer_end && gcs_file->buffer_end_is_past_eof; if (copy_size < n && !consumed_buffer_to_eof) {
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) -
tensorflow/c/experimental/next_pluggable_device/c_api.h
// Obtains key-value from coordination service agent. The returned `TF_Buffer` // is a newly allocated buffer to hold the string key-value, and caller is // responsible for managing the lifetime. If error, `status` will be set and a // nullptr will be returned. TF_CAPI_EXPORT extern TF_Buffer* TF_CoordinationServiceGetKeyValue( const char* key, int64_t key_size, TF_CoordinationServiceAgent* agent,
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Dec 20 20:01:06 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/BUILD
"//tensorflow/c:kernels_hdrs", "//tensorflow/c:tf_buffer_internal", "//tensorflow/c:tf_status_headers", "@local_xla//xla/pjrt/c:pjrt_c_api_hdrs", ], ) cc_library( name = "tensor_pjrt_buffer_util", srcs = ["tensor_pjrt_buffer_util.cc"], hdrs = ["tensor_pjrt_buffer_util.h"], visibility = ["//visibility:public"], deps = [
Plain Text - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 3.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); auto* cell2 = TFE_MonitoringGetCellStringGauge2(gauge2, "foo", "bar"); TFE_MonitoringStringGaugeCellSet(cell2, "str"); auto* buf = new TF_Buffer; TFE_MonitoringStringGaugeCellValue(cell2, buf); string data(static_cast<const char*>(buf->data), buf->length); TF_DeleteBuffer(buf); EXPECT_EQ(data, "str"); TFE_MonitoringDeleteStringGauge2(gauge2);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1)