- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 55 for tensors_ (0.35 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.h
std::vector<TFE_TensorHandle*> result; result.reserve(tensors_.size()); for (const TensorHandlePtr& tensor : tensors_) { result.emplace_back(tensor.get()); } return result; } private: ParallelTensor(const ParallelDevice& device, std::vector<TensorHandlePtr> tensors, absl::Span<const int64_t> shape, const TF_DataType dtype)
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
for (int component_index = 0; component_index < tensors_.size(); ++component_index) { // TODO(allenl): Add a C API for summarizing tensors. Currently custom // devices limiting themselves to a C API (for ABI compatibility) would need // to implement summarization for component tensors themselves. ImmediateExecutionTensorHandle* component = tensorflow::unwrap(tensors_[component_index].get());
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
}); } bool Tape::ShouldRecord( absl::Span<const AbstractTensorHandle* const> tensors) const { std::vector<int64_t> tensor_ids(tensors.size()); std::vector<tensorflow::DataType> tensor_dtypes(tensors.size()); for (int i = 0; i < tensors.size(); i++) { tensor_ids[i] = ToId(tensors[i]); tensor_dtypes[i] = tensors[i]->DataType(); } return GradientTape::ShouldRecord(tensor_ids, tensor_dtypes); }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/c/eager/gradients.h
const string& op_name = ""); // Returns whether any tensor in a list of tensors is being watched and has // a trainable dtype. bool ShouldRecord( absl::Span<const AbstractTensorHandle* const> tensors) const; // Unwatches this tensor on the tape. Mainly used for cleanup when deleting // eager tensors. void DeleteTrace(const AbstractTensorHandle*);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.h
// an error. // // All component tensors must have the same dtype. Currently they must also have // the same shape, although this requirement may be relaxed in the future. // // `device_name` must not name an existing physical or custom device (see // the documentation for TFE_RegisterCustomDevice for more information). // // Tensors may be copied on or off the device explicitly using
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Jun 04 21:49:16 UTC 2020 - 2.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
TFE_TensorHandle* tensor, TF_Status* status, void* device_info) { TF_SetStatus( status, TF_UNIMPLEMENTED, absl::StrCat("Trying to copy a tensor ", tensorflow::unwrap(tensor)->DebugString(), " on to a parallel device. Pack non-parallel " "tensors for each device into a parallel tensor explicitly.")
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.h
// some input tensors are not on that device. // LINT.IfChange // Note: Keep in sync with internal copy of enum in eager/context.h. typedef enum TFE_ContextDevicePlacementPolicy { // Running operations with input tensors on the wrong device will fail. TFE_DEVICE_PLACEMENT_EXPLICIT = 0, // Copy the tensor to the right device but log a warning. TFE_DEVICE_PLACEMENT_WARN = 1,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/c/c_api.cc
// Input tensors const std::vector<std::pair<string, Tensor>>& input_pairs, // Output tensors const std::vector<string>& output_tensor_names, TF_Tensor** c_outputs, // Target nodes const std::vector<string>& target_oper_names, TF_Buffer* run_metadata, TF_Status* status) { const int noutputs = output_tensor_names.size(); std::vector<Tensor> outputs(noutputs); Status result;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 102.3K bytes - Viewed (0) -
RELEASE.md
`strategy.scope()`. * `TensorRT` * [TensorRT 6.0](https://developer.nvidia.com/tensorrt#tensorrt-whats-new) is now supported and enabled by default. This adds support for more TensorFlow ops including Conv3D, Conv3DBackpropInputV2, AvgPool3D, MaxPool3D, ResizeBilinear, and ResizeNearestNeighbor. In addition, the TensorFlow-TensorRT python conversion API is exported as
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Tue Oct 22 14:33:53 UTC 2024 - 735.3K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_context.h
// Note: Keep in sync with exported copy of enum in eager/c_api.h. enum ContextDevicePlacementPolicy { // Running operations with input tensors on the wrong device will fail. DEVICE_PLACEMENT_EXPLICIT = 0, // Copy the tensor to the right device but log a warning. DEVICE_PLACEMENT_WARN = 1, // Silently copy the tensor, which has a performance cost since the operation // will be blocked till the copy completes. This is the default policy.
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 12.3K bytes - Viewed (0)