Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 for num_inputs (0.07 sec)

  1. tensorflow/c/c_api_experimental.cc

      using tensorflow::shape_inference::ShapeHandle;
    
      const int num_inputs = input_shapes->num_items;
      NodeDef node_def;
      tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op);
      node_def.set_name(op->Name());
      node_def.set_op(op->Name());
      for (int i = 0; i < num_inputs; ++i) {
        node_def.add_input("dummy_input");
      }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/custom_device_testutil.cc

      TFE_OpAddAttrs(op, attributes);
      TFE_OpSetDevice(op, dev->underlying_device.c_str(), s);
      if (TF_GetCode(s) != TF_OK) return;
      int num_inputs = TFE_OpGetFlatInputCount(original_op, s);
      if (TF_GetCode(s) != TF_OK) return;
      for (int j = 0; j < num_inputs; ++j) {
        TFE_TensorHandle* input = TFE_OpGetFlatInput(original_op, j, s);
        if (TF_GetCode(s) != TF_OK) return;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Mar 03 20:47:31 UTC 2021
    - 8.3K bytes
    - Viewed (0)
  3. 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: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device.cc

          reinterpret_cast<NamedParallelDevice*>(device_info);
      std::vector<MaybeParallelTensorUnowned> typed_inputs;
      int num_inputs = TFE_OpGetFlatInputCount(original_op, status);
      if (TF_GetCode(status) != TF_OK) return;
      typed_inputs.reserve(num_inputs);
      for (int i = 0; i < num_inputs; ++i) {
        TFE_TensorHandle* input = TFE_OpGetFlatInput(original_op, i, status);
        if (TF_GetCode(status) != TF_OK) return;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_unified_experimental.h

    // returned through the provided TF_OutputList.
    // Any active tape will observe the effects of this execution.
    void TF_ExecuteOperation(TF_AbstractOp* op, int num_inputs,
                             TF_AbstractTensor* const* inputs, TF_OutputList* o,
                             TF_Status* s);
    
    // Creates a new TF_AbstractFunction from the current tracing states in the
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sun Oct 24 11:27:00 UTC 2021
    - 7K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.h

    TF_CAPI_EXPORT extern void TFE_OpAddInputList(TFE_Op* op,
                                                  TFE_TensorHandle** inputs,
                                                  int num_inputs,
                                                  TF_Status* status);
    
    // Fetches the current number of inputs attached to `op`.
    //
    // Does not use the operation's definition to determine how many inputs should
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  7. tensorflow/c/c_api.cc

    }
    
    void TF_AddInputList(TF_OperationDescription* desc, const TF_Output* inputs,
                         int num_inputs) {
      std::vector<NodeBuilder::NodeOut> input_list;
      input_list.reserve(num_inputs);
      for (int i = 0; i < num_inputs; ++i) {
        input_list.emplace_back(&inputs[i].oper->node, inputs[i].index);
      }
      desc->node_builder.Input(input_list);
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api.cc

    }
    
    void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs,
                            TF_Status* status) {
      status->status = tensorflow::unwrap(op)->AddInputList(
          {reinterpret_cast<tensorflow::AbstractTensorHandle**>(
               tensorflow::unwrap(inputs)),
           static_cast<size_t>(num_inputs)});
    }
    
    extern int TFE_OpGetFlatInputCount(const TFE_Op* op, TF_Status* status) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  9. tensorflow/c/c_api.h

    // For inputs that take a list of tensors.
    // inputs must point to TF_Output[num_inputs].
    TF_CAPI_EXPORT extern void TF_AddInputList(TF_OperationDescription* desc,
                                               const TF_Output* inputs,
                                               int num_inputs);
    
    // Call once per control input to `desc`.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_test.cc

      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpAddAttrs(ret, TFE_OpGetAttrs(other));
      int num_inputs = TFE_OpGetFlatInputCount(other, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      for (int input_index = 0; input_index < num_inputs; ++input_index) {
        TFE_TensorHandle* input = TFE_OpGetFlatInput(other, input_index, status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
Back to top