Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for num_inputs (0.13 sec)

  1. tensorflow/c/experimental/gradients/grad_test_helper.cc

        double abs_error) {
      auto num_inputs = inputs.size();
      std::vector<AbstractTensorHandle*> outputs(num_inputs);
      auto s = RunModel(grad_model, ctx, inputs, absl::MakeSpan(outputs),
                        /*use_function=*/use_function);
      ASSERT_EQ(errors::OK, s.code()) << s.message();
    
      for (int i = 0; i < num_inputs; ++i) {
        if (!outputs[i]) continue;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 28 13:53:47 UTC 2024
    - 5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental.cc

                   absl::StatusMessageAsCStr(status));
    }
    
    void TF_ExecuteOperation(TF_AbstractOp* op, int num_inputs,
                             TF_AbstractTensor* const* inputs, TF_OutputList* o,
                             TF_Status* s) {
      for (int i = 0; i < num_inputs; i++) {
        tsl::Set_TF_Status_from_Status(s, unwrap(op)->AddInput(unwrap(inputs[i])));
        if (TF_GetCode(s) != TF_OK) {
          return;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 10:15:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op.cc

            absl::AbortedError(
                "The `ids` and `calibration_methods` must have the same size."));
    
        // Check the number and type of inputs.
        OP_REQUIRES(context, context->num_inputs() == ids_.size() * 3,
                    absl::AbortedError("The number of inputs must be  three times "
                                       "the size of the `ids` list."));
        for (int i = 0; i < ids_.size(); ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 13 01:31:23 UTC 2024
    - 8K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/variable_info_util.cc

      }
      return absl::OkStatus();
    }
    
    std::vector<int> GetResourceVariableIndicesFromContext(OpKernelContext* ctx) {
      std::vector<int> out;
      for (int64 i = 0; i < ctx->num_inputs(); i++) {
        if (ctx->input(i).dtype() == DT_RESOURCE) {
          out.push_back(i);
        }
      }
      return out;
    }
    
    Status CreateVariableInfoLookup(
        absl::Span<VariableInfo const> variable_args,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 7K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/ops/gen/cpp/views/op_view.cc

    const std::vector<AttrView>& OpView::Attributes() const {
      return argument_attrs_;
    }
    
    const std::vector<OpArgumentView>& OpView::AllArguments() const {
      return all_arguments_;
    }
    
    int OpView::NumInputs() const { return input_args_.size(); }
    
    int OpView::NumOutputs() const { return output_args_.size(); }
    
    ArgView OpView::OnlyInput() const {
      CHECK_EQ(input_args_.size(), 1);  // Crash OK
      return input_args_.front();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 03 07:02:00 UTC 2024
    - 3.2K bytes
    - Viewed (0)
Back to top