Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for ScalarsFromSequence (0.21 sec)

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

      std::unique_ptr<ParallelTensor> reduced_values =
          parallel_device.ScalarsFromSequence<float>({1.0, 2.0}, context.get(),
                                                     status.get());
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
      std::unique_ptr<ParallelTensor> run_collective =
          parallel_device.ScalarsFromSequence<bool>({true, true}, context.get(),
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 15.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_lib.h

      // Construct a parallel tensor consisting of the scalar values from `values`.
      template <typename DataType>
      std::unique_ptr<ParallelTensor> ScalarsFromSequence(
          absl::Span<const DataType> values, TFE_Context* context,
          TF_Status* status) const;
    
      // A parallel tensor with scalar integers numbering component devices.
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Apr 25 15:21:13 GMT 2023
    - 12.9K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      std::vector<int32_t> ids;
      ids.reserve(num_underlying_devices());
      for (int i = 0; i < num_underlying_devices(); ++i) {
        ids.push_back(i);
      }
      return ScalarsFromSequence<int32_t>(ids, context, status);
    }
    
    absl::optional<std::vector<std::unique_ptr<ParallelTensor>>>
    ParallelDevice::Execute(TFE_Context* context,
                            const std::vector<ParallelTensor*>& inputs,
    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)
Back to top