Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 6 of 6 for sum_output (0.05 seconds)

  1. 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
    
    absl::Status GradientRegistry::Register(
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Tue Feb 24 06:18:31 GMT 2026
    - 19.6K bytes
    - Click Count (0)
  2. 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) {
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat May 31 07:13:41 GMT 2025
    - 15.7K bytes
    - Click Count (0)
  3. 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, "'");
            }
          }
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 16:27:48 GMT 2024
    - 13.7K bytes
    - Click Count (0)
  4. tensorflow/c/c_api_experimental.cc

      status->status = c.Run(op_reg_data->shape_inference_fn);
      if (!status->status.ok()) return;
    
      // Set output_shapes.
      TF_ShapeAndTypeList* output_shapes_result =
          TF_NewShapeAndTypeList(c.num_outputs());
      for (int i = 0; i < c.num_outputs(); ++i) {
        ShapeHandle shape_handle = c.output(i);
        TF_ShapeAndType& shape = output_shapes_result->items[i];
        shape.num_dims = c.Rank(shape_handle);
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 04 05:55:32 GMT 2025
    - 29.4K bytes
    - Click Count (0)
  5. tensorflow/c/eager/c_api_unified_experimental_test.cc

        add_output2 = TF_OutputListGet(add_outputs, 0);
        TF_DeleteOutputList(add_outputs);
      }
    
      // 3rd Output will be Matrix Multiplication of add_output1 and add_output2
      TF_AbstractTensor* mm_output;
      {
        // Build an abstract operation, inputs and output.
        auto* mm_op = TF_NewAbstractOp(graph_ctx);
        TF_AbstractOpSetOpType(mm_op, "MatMul", s);
        ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 39.1K bytes
    - Click Count (0)
  6. tensorflow/c/eager/c_api_experimental.h

      // device if any of their inputs are on that custom device, but custom devices
      // are free to set a bad status in order to require explicit placement.
      void (*execute)(const TFE_Op* op, int* num_outputs,
                      TFE_TensorHandle** outputs, TF_Status* s, void* device_info);
    
      // Method to delete a device.
      void (*delete_device)(void* device_info);
    
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Click Count (1)
Back to Top