Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 41 for be (0.14 sec)

  1. tensorflow/c/eager/dlpack.cc

    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;
      std::vector<int64_t> strides;
      DLManagedTensor tensor;
    
    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)
  2. tensorflow/c/env_test.cc

        const char* childpath;
        ASSERT_TRUE(TF_StringStreamNext(children, &childpath));
        ASSERT_EQ(::tensorflow::string(childpath), "somefile.txt");
        // There should only be one file in this directory.
        ASSERT_FALSE(TF_StringStreamNext(children, &childpath));
        ASSERT_EQ(childpath, nullptr);
        TF_StringStreamDone(children);
    
        TF_FileStatistics stats;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Dec 10 20:52:48 GMT 2018
    - 4.2K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients.cc

     private:
      // The context where the aggregation op `Add` is to be created.
      AbstractContext* ctx_;
    };
    
    // Returns the number of elements in the gradient tensor.
    int64_t TapeVSpace::NumElements(AbstractTensorHandle* tensor) const {
      // 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
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device.cc

            // copies/broadcasts. Other implicit copies may be supported eventually,
            // but need special handling for gradients (gradient of copy-on is not
            // just copy-off but includes a sum) and consideration of performance.
            //
            // TODO(allenl): There may be smarter ways to do this copy in some
            // cases, i.e. with a collective broadcast. We'll need to be careful
    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)
  5. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

    }  // namespace
    
    // Allows a single op at a time to be launched without blocking.
    //
    // DeviceThread itself is thread-safe, in that StartExecute will block if there
    // is a pending execution. Since StartExecute is equivalent to grabbing a lock,
    // multiple DeviceThreads should always be accessed in the same order to avoid
    // deadlocks.
    class DeviceThread {
     public:
    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)
  6. tensorflow/c/experimental/grappler/grappler_test.cc

      tensorflow::Status status = InitGraphPlugin(plugin_init);
      ASSERT_EQ(status.code(), tensorflow::error::FAILED_PRECONDITION);
      ASSERT_EQ(
          status.message(),
          "'device_type' field in TP_OptimizerRegistrationParams must be set.");
    }
    
    TEST(Grappler, OptimizeFuncNotSet) {
      auto plugin_init = [](TP_OptimizerRegistrationParams* const params,
                            TF_Status* const status) -> void {
        TF_SetStatus(status, TF_OK, "");
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Thu Apr 13 22:30:58 GMT 2023
    - 11.6K bytes
    - Viewed (0)
  7. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      TensorHandlePtr cpu_value(FloatTensorHandle(3., status.get()));
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    
      // Copying on to a parallel device must be explicit.
      TensorHandlePtr failed_copy_on_result(TFE_TensorHandleCopyToDevice(
          cpu_value.get(), context.get(), device_name, status.get()));
      EXPECT_EQ(TF_GetCode(status.get()), TF_UNIMPLEMENTED);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  8. tensorflow/c/eager/custom_device_testutil.cc

      LoggingDevice* dev = reinterpret_cast<LoggingDevice*>(device_info);
      if (dev->strict_scope_placement && *requested_placement == '\0') {
        TF_SetStatus(s, TF_INTERNAL,
                     "Ops must be placed on the device explicitly, or their inputs "
                     "first copied to other devices.");
        return;
      }
      TFE_Context* context = TFE_OpGetContext(original_op, s);
      if (TF_GetCode(s) != TF_OK) return;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 03 20:47:31 GMT 2021
    - 8.3K bytes
    - Viewed (0)
  9. tensorflow/c/eager/parallel_device/parallel_device_testlib.cc

                          TF_Status* status) {
      GeneralAssignment("AssignVariableOp", context, value, status);
    }
    
    // Passed to `TF_NewTensor` to indicate how an array of floats should be
    // deleted.
    static void FloatDeallocator(void* data, size_t, void* arg) {
      delete[] static_cast<float*>(data);
    }
    
    // Creates a TFE_TensorHandle with value `v`.
    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)
  10. tensorflow/c/experimental/next_pluggable_device/c_api.cc

    }
    
    // -------------------------  VariableInfo  ------------------------------------
    struct TF_VariableInfo {
      TF_VariableInfo() = delete;
      // TF_VariableInfo is constructed here by TensorFlow, and will be passed to
      // plugin as a opaque pointer. Plugin will need to call C APIs below to
      // operate on TF_VariableInfo (such as allocate temp tensor for the `var` held
      // by the underlying tensorflow::VariableInfo.
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jan 09 00:52:04 GMT 2024
    - 13.9K bytes
    - Viewed (1)
Back to top