- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for Chen (0.18 sec)
-
tensorflow/c/c_api.cc
if (prefix != nullptr && !absl::StartsWith(n->name(), prefix_cmp)) { status->status = tensorflow::errors::Internal( "BUG: The gradients prefix have been unexpectedly altered when " "adding the nodes to the graph. This is a bug. Please file an " "issue at https://github.com/tensorflow/tensorflow/issues."); return; }
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/eager/c_api_distributed_test.cc
TFE_ContextAsyncWait(ctx, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); // Pack 3 variable handles into one TFE_TensorHandle. // When remote is false, function device is placed on task0. Handle types are // REMOTE, REMOTE, LOCAL on task0. When remote is true, function device is // placed on task1, Handle types are LOCAL, REMOTE, LOCAL on task1. int num_replicas = 3;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
} else { // This node is referenced in inputs. Currently, we place an // artificial restriction and require that when num_opers=-1, such // nodes must have a single output. if (node->num_outputs() != 1) { return InvalidArgument( "When `num_opers` is set to -1, nodes referenced in `inputs` " "must have a single output. Node ",
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/eager/gradient_checker.cc
TF_Tensor* model_out_tensor; TF_RETURN_IF_ERROR(GetValue(model_out.get(), &model_out_tensor)); int num_dims_out = TF_NumDims(model_out_tensor); TF_DeleteTensor(model_out_tensor); // If the output is a scalar, then return the scalar output if (num_dims_out == 0) { outputs[0] = model_out.release(); return absl::OkStatus(); } // Else, reduce sum the output to get a scalar
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
PartialTensorShape first_shape; TF_RETURN_IF_ERROR(unwrap(tensors_[0].get())->Shape(&first_shape)); return errors::Unimplemented(absl::StrCat( "Computing the shape of a ParallelTensor when the components do " "not all have the same rank is not supported. One tensor had " "shape ", first_shape.DebugString(), " and another had shape ",
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/c_api_experimental.cc
// function. Otherwise return nullptr in result and error Status from the // function. // // If `library_filename` has already been loaded, we return a cached handle. // Device and Kernels/Ops are registered as globals when a library is loaded // for the first time. TF_Library* TF_LoadPluggableDeviceLibrary(const char* library_filename, TF_Status* status) {
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/c_api_unified_experimental_test.cc
using tensorflow::TF_StatusPtr; namespace tensorflow { namespace { // The tests are parameterized on: // - a string representing the tracing implementation: "mlir" or "graphdef". // - a boolean that when true enables TFRT as the execution engine. class UnifiedCAPI : public ::testing::TestWithParam<std::tuple<const char*, bool>> { protected: void SetUp() override { TF_StatusPtr status(TF_NewStatus());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
} int num_retvals = 1; if (async) { // Enqueue dummy ops so we backlog async execution & actually test async. // This is usually unnecessary, but we've experienced the occasional test // failure when testing async mode with no explicit forwarding. for (int i = 0; i < 100000; ++i) { TFE_Op* add_op_dummy = AddOp(ctx, m, m); TFE_OpSetDevice(add_op_dummy, cpu_device_name.c_str(), status);
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/dlpack.cc
#include "tensorflow/core/platform/logging.h" namespace tensorflow { namespace { // Managing context for the DLManagedTensor, will manage the lifetime of // DLManagedTensor. When calling DLManagedTensor::deleter, it will notify the // original framework of destruction, and this context will be deleted also. struct TfDlManagedTensorCtx { TensorReference reference; std::vector<int64_t> shape;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
constexpr char kStatCacheMaxEntries[] = "GCS_STAT_CACHE_MAX_ENTRIES"; constexpr size_t kStatCacheDefaultMaxEntries = 1024; // How to upload new data when Flush() is called multiple times. // By default the entire file is reuploaded. constexpr char kAppendMode[] = "GCS_APPEND_MODE"; // If GCS_APPEND_MODE=compose then instead the new data is uploaded to a // temporary object and composed with the original object. This is disabled by
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)