Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 21 for We (0.19 sec)

  1. tensorflow/c/eager/dlpack.cc

      // There are two ways to represent compact row-major data
      // 1) nullptr indicates tensor is compact and row-majored.
      // 2) fill in the strides array as the real case for compact row-major data.
      // Here we choose option 2, since some frameworks didn't handle the strides
      // argument properly.
      dlm_tensor->dl_tensor.strides = stride_arr->data();
    
      dlm_tensor->dl_tensor.byte_offset =
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.cc

      // TODO(srbs): It seems like this is used only for performance optimization
      // and not for correctness. The only downside of keeping this 1 seems to be
      // that the gradient accumulation is unbounded and we will never
      // aggressively aggregate accumulated gradients to recover memory.
      // Revisit and fix.
      return 1;
    }
    
    // Consumes references to the tensors in the gradient_tensors list and returns
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

        // Wait until all pending nodes have completed since they may have a
        // reference to default_cancellation_manager_. We ignore the status return
        // since we already have a bad status to propagate.
        TFE_ContextAsyncWait(context, await_status.get());
        // Reset the cancellation manager on a bad status. Otherwise we'll cancel
        // all future operations.
        default_cancellation_manager_ = absl::make_unique<CancellationManager>();
      }
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  4. tensorflow/c/c_api_experimental.cc

      gpu_options->set_allow_growth(gpu_memory_allow_growth);
    
      (*config.mutable_device_count())["CPU"] = num_cpu_devices;
    
      // TODO(b/113217601): This is needed for EagerContext::runner_ to use a
      // threadpool, so that we avoid the possibility of running the runner_ in the
      // threadpool of GPU event mgr, as that can trigger more callbacks to be
      // scheduled on that same threadpool, causing a deadlock in cases where the
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  5. tensorflow/c/eager/parallel_device/parallel_device.cc

        const TFE_OpAttrs* attributes, int expected_max_outputs,
        TF_Status* status) {
      absl::optional<std::vector<MaybeParallelTensorOwned>> result;
      // 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")) {
    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)
  6. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc

          TF_SetStatusFromIOError(status, errno, dst);
          return;
        }
      } else if (S_ISDIR(st.st_mode)) {
        TF_SetStatus(status, TF_FAILED_PRECONDITION, "target path is a directory");
        return;
      }
    
      // We cannot rename directories yet, so prevent this.
      if (stat(src, &st) != 0) {
        TF_SetStatusFromIOError(status, errno, src);
        return;
      } else if (S_ISDIR(st.st_mode)) {
    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)
  7. tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem_test.cc

    #define ASSERT_TF_OK(x) ASSERT_EQ(TF_OK, TF_GetCode(x)) << TF_Message(x)
    #define EXPECT_TF_OK(x) EXPECT_EQ(TF_OK, TF_GetCode(x)) << TF_Message(x)
    
    static const char* content = "abcdefghijklmnopqrstuvwxyz1234567890";
    // We will work with content_view instead of content.
    static const absl::string_view content_view = content;
    
    namespace gcs = google::cloud::storage;
    
    static std::string InitializeTmpDir() {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Aug 31 12:04:23 GMT 2020
    - 24.9K bytes
    - Viewed (0)
  8. tensorflow/c/eager/parallel_device/parallel_device_testlib.cc

      TFE_OpSetAttrShape(op.get(), "shape", dims, num_dims, status);
      TFE_OpSetAttrString(op.get(), "container", "", 0);
      // 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";
    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)
  9. tensorflow/c/eager/c_api_cluster_test.cc

    TEST(CAPI, TestRemoteExecuteUpdateServerDefResourceAccessAsync) {
      TestRemoteExecuteUpdateServerDefResourceAccess(true);
    }
    
    void TestRemoteExecuteUpdateServerDefWithFailures(bool async) {
      // Fail fast on GetStatus requests so we can get errors instead of timeout
      // when updating cluster with non-exsitent worker
      tensorflow::setenv("GRPC_FAIL_FAST", "TRUE", /*overwrite=*/1);
    
      tensorflow::ServerDef server_def = GetServerDef(2);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  10. tensorflow/c/experimental/filesystem/modular_filesystem_registration.cc

      return OkStatus();
    }
    
    // Validates the random access file operations supplied by the plugin.
    static Status ValidateHelper(const TF_RandomAccessFileOps* ops) {
      if (ops == nullptr) {
        // We allow filesystems where files can only be written to (from TF code)
        return OkStatus();
      }
    
      if (ops->cleanup == nullptr)
        return errors::FailedPrecondition(
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 07 22:08:43 GMT 2023
    - 12.8K bytes
    - Viewed (0)
Back to top