Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for TF_AddInputList (0.11 sec)

  1. tensorflow/c/c_test_util.cc

                     TF_Status* s, const char* name, TF_Operation** op,
                     bool check) {
      TF_OperationDescription* desc = TF_NewOperation(graph, "AddN", name);
      TF_Output add_inputs[2] = {{l, 0}, {r, 0}};
      TF_AddInputList(desc, add_inputs, 2);
      *op = TF_FinishOperation(desc, s);
      if (check) {
        ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        ASSERT_NE(*op, nullptr);
      }
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.h

    // single tensor), and TF_AddInputList() for the second input (since
    // it takes a list, even if you were to pass a list with a single
    // tensor), as in:
    //   TF_OperationDescription* desc = TF_NewOperation(graph, "Concat", "c");
    //   TF_Output concat_dim_input = {...};
    //   TF_AddInput(desc, concat_dim_input);
    //   TF_Output values_inputs[5] = {{...}, ..., {...}};
    //   TF_AddInputList(desc, values_inputs, 5);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_unified_experimental_graph.cc

          if (!t) {
            return tensorflow::errors::InvalidArgument(
                "Unable to cast input to GraphTensor");
          }
          tf_outputs[i] = t->output_;
        }
        TF_AddInputList(op_.get(), tf_outputs.data(), tf_outputs.size());
        return absl::OkStatus();
      }
      absl::Status Execute(absl::Span<AbstractTensorHandle*> retvals,
                           int* num_retvals) override {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api.h

    // TFE_DeleteOp() is called before TFE_DeleteContext().
    //
    // Very similar to TF_OperationDescription with some differences:
    // (1) TF_Output or TFE_TensorHandle* as arguments to TF_AddInput,
    //     TF_AddInputList
    // (2) TF_ColocateWith, TF_AddControlInput etc. do not make sense.
    // (3) Implementation detail: Avoid use of NodeBuilder/NodeDefBuilder since
    //     the additional sanity checks there seem unnecessary;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  5. tensorflow/c/c_api.cc

      desc->node_builder.Device(device);
    }
    
    void TF_AddInput(TF_OperationDescription* desc, TF_Output input) {
      desc->node_builder.Input(&input.oper->node, input.index);
    }
    
    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) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_test.cc

        ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
        desc_ = TF_NewOperation(graph_, "AddN", "add");
        TF_Output inputs[] = {{feed1_, 0}, {constant_, 0}};
        TF_AddInputList(desc_, inputs, TF_ARRAYSIZE(inputs));
      }
    
      ~CApiColocationTest() override {
        TF_DeleteGraph(graph_);
        TF_DeleteStatus(s_);
      }
    
      void SetViaStringList(TF_OperationDescription* desc,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
Back to top