- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for spacing (0.19 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
// Use the special GUID for no buffer sharing // // TODO(allenl): Should we provide a better API for this? AFAIK this is the // only reasonable way to make variables with no aliasing using the eager C // API. std::string no_sharing = "cd2c89b7-88b7-44c8-ad83-06c2a9158347"; TFE_OpSetAttrString(op.get(), "shared_name", no_sharing.c_str(), no_sharing.length()); TFE_OpSetDevice(op.get(), device, status);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
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 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K 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 23 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_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 23 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 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)