- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for racking (0.22 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.cc
// TODO(allenl): We should remove "TPU" from these op names at the very least, // or consider other ways of packing/unpacking parallel tensors. if (operation_name == std::string("TPUReplicatedInput")) { // Special-cased operation for packing per-device tensors into one parallel // tensor. if (inputs.size() != parallel_device.num_underlying_devices()) { std::string message(absl::StrCat(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
if (TF_GetCode(status) != TF_OK) return nullptr; return new Variable(var_handle, type); } void Variable::Destroy(TFE_Context* context, TF_Status* status) { // Free the backing buffer for the variable. std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( TFE_NewOp(context, "DestroyResourceOp", status), &TFE_DeleteOp); if (TF_GetCode(status) != TF_OK) return;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache.cc
bool downloaded_block = false; auto reconcile_state = MakeCleanup([this, &downloaded_block, &key, &block] { // Perform this action in a cleanup callback to avoid locking mu_ after // locking block->mu. if (downloaded_block) { absl::MutexLock l(&mu_); // Do not update state if the block is already to be evicted. if (block->timestamp != 0) {
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/eager/c_api_test.cc
// .backing_device of shape is CPU since the tensor is backed by CPU backing_device_name = TFE_TensorHandleBackingDeviceName(retvals[0], status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); ASSERT_TRUE(absl::StrContains(backing_device_name, "CPU:0")) << backing_device_name; TFE_DeleteOp(shape_op);
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/eager/custom_device_test.cc
TF_DeleteTensor); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); ASSERT_EQ(111., *static_cast<float*>(TF_TensorData(resolved_value.get()))); // Free the backing buffer for the variable. op.reset(TFE_NewOp(context.get(), "DestroyResourceOp", status.get())); TFE_OpAddInput(op.get(), var_handle, status.get()); TFE_OpSetDevice(op.get(), name, status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/eager/tracing_utils.cc
#include "tensorflow/c/eager/tracing_utils.h" #include "tensorflow/c/eager/c_api_unified_experimental_internal.h" #include "tensorflow/c/experimental/gradients/tape/tape_operation.h" #include "tensorflow/core/lib/llvm_rtti/llvm_rtti.h" #include "tensorflow/core/platform/errors.h" namespace tensorflow { namespace tracing { Status MaybeSetOpName(AbstractOperation* op, const char* op_name) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test_util.cc
ASSERT_TRUE(remote_arg->HasLocalMirror(nullptr)); } if (remote_func_outputs) { const string backing_device = TFE_TensorHandleBackingDeviceName(retvals[0], status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); EXPECT_EQ(backing_device, task2_name); } auto* retval_task0 = TFE_TensorHandleCopyToDevice(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Dec 11 22:56:03 GMT 2020 - 9.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) { // One of the inputs we're trying to pack is on a custom device. We'll let // the first custom device we see handle all of the packing. auto* custom_device_handle = tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( unwrapped_handle); tensorflow::ImmediateExecutionTensorHandle* result;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/eager/c_api.cc
void* device_info) { TF_SetStatus(status, TF_UNIMPLEMENTED, "This custom device does not support packing tensors."); return nullptr; } } // namespace extern "C" { bool TFE_IsCustomDevice(TFE_Context* ctx, const char* device_name) { return tensorflow::unwrap(ctx)->IsCustomDevice(device_name);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
if (read_fill_buffer >= 0) gcs_file->buffer.resize(read_fill_buffer); if (TF_GetCode(status) != TF_OK && TF_GetCode(status) != TF_OUT_OF_RANGE) { // Empty the buffer to avoid caching bad reads. gcs_file->buffer.resize(0); return -1; } size_t remaining_copy = (std::min)(n - copy_size, gcs_file->buffer.size());
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)