- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ScalarsFromSequence (0.17 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(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K 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.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K 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,
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)