Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 16 of 16 for AddInput (0.2 sec)

  1. 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
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
  2. tensorflow/c/c_api.h

    // You must use TF_AddInput() for the first input (since it takes a
    // 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);
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
  3. tensorflow/c/eager/c_api.cc

      return tensorflow::unwrap(op)->DeviceName().c_str();
    }
    
    void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) {
      status->status = tensorflow::unwrap(op)->AddInput(tensorflow::unwrap(input));
    }
    
    void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs,
                            TF_Status* status) {
      status->status = tensorflow::unwrap(op)->AddInputList(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  4. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      const std::string final_op("Id");
      TF_SetAttrString(reduce_desc, "final_op", final_op.c_str(),
                       final_op.length());
      TF_SetAttrIntList(reduce_desc, "subdiv_offsets", nullptr, 0);
      TF_AddInput(reduce_desc, x);
      TF_Operation* reduce_op = TF_FinishOperation(reduce_desc, status);
      if (TF_GetCode(status) != TF_OK) return;
      TF_Operation* operations[]{placeholder_op, reduce_op};
      TF_Output y{reduce_op, 0};
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  5. tensorflow/c/c_api.cc

      return TF_NewOperationLocked(graph, op_type, oper_name);
    }
    
    void TF_SetDevice(TF_OperationDescription* desc, const char* device) {
      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,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_function_test.cc

      void UseHelper(const std::vector<TF_Output>& inputs, TF_Operation** op) {
        TF_OperationDescription* desc =
            TF_NewOperation(host_graph_, func_name_, func_node_name_);
        for (auto input : inputs) {
          TF_AddInput(desc, input);
        }
        // Set device to CPU because some ops inside the function might not be
        // available on GPU.
        TF_SetDevice(desc, "/cpu:0");
        *op = TF_FinishOperation(desc, s_);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
Back to top