- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for summarizer (0.16 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_test.cc
ImmediateExecutionTensorHandle* unwrapped_handle = tensorflow::unwrap(device_value.get()); std::string summarized; TF_ASSERT_OK(unwrapped_handle->SummarizeValue(summarized)); EXPECT_THAT(summarized, HasSubstr("\"CPU:0\": 3")); } } // namespace parallel_device
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) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
} *shape = &*shape_; return absl::OkStatus(); } Status ParallelTensor::SummarizeValue(std::string& summary) { summary = "{"; std::vector<std::string> summarized_devices = device_.SummarizeDeviceNames(); for (int component_index = 0; component_index < tensors_.size(); ++component_index) { // TODO(allenl): Add a C API for summarizing tensors. Currently custom
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) -
tensorflow/c/eager/parallel_device/parallel_device.cc
TFE_CustomDeviceTensorHandleMethods handle_methods; handle_methods.num_dims = &ParallelTensorNumDims; handle_methods.dim = &ParallelTensorDim; handle_methods.deallocator = &ParallelTensorDeallocator; handle_methods.summarize = &ParallelTensorSummarize; return TensorHandlePtr(TFE_NewCustomDeviceTensorHandle( context, parallel_device_name.c_str(), t_released->dtype(), t_released, handle_methods, status)); }
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) -
tensorflow/c/eager/c_api.cc
return methods_.summarize != nullptr; } Status SummarizeValue(std::string& summary) const override { if (methods_.summarize == nullptr) { return tensorflow::CustomDeviceTensorHandle::SummarizeValue(summary); } TF_Status c_status; std::unique_ptr<TF_Buffer, decltype(&TF_DeleteBuffer)> summary_buffer( methods_.summarize(data_, &c_status), TF_DeleteBuffer);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)