Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for num_outputs (0.19 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top