- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ScalarsFromSequence (0.12 sec)
-
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(),
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 15.6K bytes - Viewed (0) -
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.
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 13.1K bytes - Viewed (0) -
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,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 25.9K bytes - Viewed (0)