- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for num_outputs (0.19 sec)
-
tensorflow/c/eager/c_api_unified_experimental.cc
void TF_DeleteOutputList(TF_OutputList* o) { delete unwrap(o); } void TF_OutputListSetNumOutputs(TF_OutputList* o, int num_outputs, TF_Status* s) { unwrap(o)->expected_num_outputs = num_outputs; unwrap(o)->outputs.clear(); unwrap(o)->outputs.resize(num_outputs); } int TF_OutputListNumOutputs(TF_OutputList* o) { return unwrap(o)->outputs.size(); }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
absl::StrCat("ZerosLike", ToId(t)).c_str())); } TF_RETURN_IF_ERROR(op->AddInput(t)); int num_outputs = 1; std::vector<AbstractTensorHandle*> outputs(num_outputs); TF_RETURN_IF_ERROR( op->Execute(absl::Span<AbstractTensorHandle*>(outputs), &num_outputs)); *result = outputs[0]; return absl::OkStatus(); } } // namespace Status GradientRegistry::Register(
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
} std::vector<TFE_TensorHandle*> op_outputs(*num_outputs); TFE_Execute(op, op_outputs.data(), num_outputs, s); TFE_DeleteOp(op); if (TF_GetCode(s) != TF_OK) return; std::vector<TFE_TensorHandle*> unwrapped_outputs; unwrapped_outputs.reserve(op_outputs.size()); for (auto* handle : op_outputs) { unwrapped_outputs.push_back(handle); } for (int i = 0; i < *num_outputs; ++i) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
shape.dims())); } int num_outputs = 1; std::vector<AbstractTensorHandle*> outputs(num_outputs); TF_RETURN_IF_ERROR(operation->Execute( absl::Span<AbstractTensorHandle*>(outputs), &num_outputs)); if (num_outputs != 1) { return errors::Internal("Expected 1 output but found ", num_outputs); } auto* t = dyn_cast<GraphTensor>(outputs[0]); if (!t) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/eager/unified_api_testutil.cc
AbstractFunction* func = nullptr; OutputList output_list; output_list.expected_num_outputs = 0; output_list.outputs.reserve(outputs.size()); for (int i = 0; i < model_outputs.size(); i++) { if (model_outputs[i]) { output_list.outputs.emplace_back(model_outputs[i]); output_list.expected_num_outputs += 1; } else { null_indices.insert(i); } }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
attributes, *num_outputs, status)); if (TF_GetCode(status) != TF_OK) return; if (!maybe_typed_outputs.has_value()) { TF_SetStatus(status, TF_INTERNAL, "OK status but no value was returned."); return; } std::vector<MaybeParallelTensorOwned> typed_outputs( std::move(maybe_typed_outputs.value())); if (typed_outputs.size() > *num_outputs) { TF_SetStatus(status, TF_INTERNAL,
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
} TFE_Execute(op_.get(), unwrapped_results.data(), &real_num_outputs, status); if (TF_GetCode(status) != TF_OK) { cancellation_manager_->StartCancel(); return; } unwrapped_results.resize(real_num_outputs); outputs->reserve(real_num_outputs); for (TFE_TensorHandle* unwrapped_result : unwrapped_results) { outputs->emplace_back(unwrapped_result); }
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) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrType(const_op.get(), "dtype", datatype_enum); TFE_TensorHandle* device_handle; int num_outputs = 1; TFE_Execute(const_op.get(), &device_handle, &num_outputs, status); if (TF_GetCode(status) != TF_OK) return nullptr; components.emplace_back(device_handle); } return ParallelTensor::FromTensorHandles(*this, std::move(components),
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) -
tensorflow/c/eager/c_api_unified_experimental_internal.h
#include "tensorflow/core/platform/types.h" namespace tensorflow { // Represents the results of the execution of an operation. struct OutputList { std::vector<AbstractTensorHandle*> outputs; int expected_num_outputs = -1; }; namespace tracing { // ============================================================================= // Implementation detail for the unified execution APIs for Eager and tracing // backends (graph/MLIR).
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Nov 13 22:20:40 GMT 2020 - 5.2K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
// nodes must have a single output. if (node->num_outputs() != 1) { return InvalidArgument( "When `num_opers` is set to -1, nodes referenced in `inputs` " "must have a single output. Node ", node->name(), " has ", node->num_outputs(), " outputs. Encountered while creating function '", fn_name, "'"); } }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2)