Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for un (0.15 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device.cc

      if (*requested_placement == '\0') {
        TF_SetStatus(
            status, TF_INTERNAL,
            "Ops must be placed on the parallel device explicitly, or their inputs "
            "first un-packed. Got an un-placed op with an input placed on the "
            "parallel device.");
        return;
      }
      TFE_Context* context = TFE_OpGetContext(original_op, status);
      if (TF_GetCode(status) != TF_OK) return;
    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)
  2. tensorflow/c/eager/parallel_device/parallel_device.h

    // parallel device creates a parallel tensor `x` with `a` on the first of
    // `underlying_devices` and `b` on the second. Running `a_unpacked, b_unpacked =
    // TPUReplicatedOutput(input=x, num_replicas=2)` un-packs the parallel tensor
    // into its components.
    //
    // The filled `device` struct and the allocated `device_info` struct may be
    // passed to TFE_RegisterCustomDevice. The `device_name` arguments must match.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jun 04 21:49:16 GMT 2020
    - 2.9K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_testlib.h

    // Creates a rank-one TFE_TensorHandle with value `v`.
    TensorHandlePtr VectorFloatTensorHandle(const std::vector<float>& v,
                                            TF_Status* status);
    
    // Helper to un-pack `num_replicas` TFE_TensorHandles from one parallel handle.
    template <std::size_t num_replicas>
    void ExtractPerDeviceValues(
        TFE_Context* context, TFE_TensorHandle* input,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 09 01:12:35 GMT 2021
    - 6.9K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device_test.cc

          Multiply(context.get(), second_combined_value.get(),
                   second_negative_one.get(), status.get()));
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    
      // Un-pack the parallel tensor to verify that the operation was
      // successful. The resulting structure should be:
      //   second_device{first_device{1. * 3., 2. * 3.}, 3. * 3.}.
      std::array<TensorHandlePtr, 2> second_components;
    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)
  5. tensorflow/c/eager/parallel_device/parallel_device_testlib.cc

                       &FloatDeallocator, nullptr),
          TF_DeleteTensor);
      return TensorHandlePtr(TFE_NewTensorHandle(tensor.get(), status));
    }
    
    // Helper to un-pack `num_replicas` TFE_TensorHandles from one parallel handle.
    template <std::size_t num_replicas>
    void ExtractPerDeviceValues(
        TFE_Context* context, TFE_TensorHandle* input,
    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)
  6. RELEASE.md

    *   Added a versioning framework for `GraphDef`s to ensure compatibility
    *   Enforced Python 3 compatibility
    *   Internal changes now show up as sensibly separated commits
    *   Open-sourced the doc generator
    *   Un-fork Eigen
    *   Simplified the `BUILD` files and cleaned up C++ headers
    *   TensorFlow can now be used as a submodule in another bazel build
    *   New ops (e.g., `*fft`, `*_matrix_solve`)
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 29 19:17:57 GMT 2024
    - 727.7K bytes
    - Viewed (8)
Back to top