Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TF_AddControlInput (0.2 sec)

  1. tensorflow/c/c_test_util.cc

      TF_OperationDescription* desc = TF_NewOperation(graph, "AddN", name);
      TF_Output add_inputs[2] = {{l, 0}, {r, 0}};
      TF_AddInputList(desc, add_inputs, 2);
      TF_AddControlInput(desc, ctrl_op);
      return TF_FinishOperation(desc, s);
    }
    
    // If `op_device` is non-empty, set the created op on that device.
    void BinaryOpHelper(const char* op_name, TF_Operation* l, TF_Operation* r,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  2. tensorflow/c/eager/c_api.h

    //
    // 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;
    typedef struct TFE_Op TFE_Op;
    
    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)
  3. tensorflow/c/c_api.h

                                               const TF_Output* inputs,
                                               int num_inputs);
    
    // Call once per control input to `desc`.
    TF_CAPI_EXPORT extern void TF_AddControlInput(TF_OperationDescription* desc,
                                                  TF_Operation* input);
    
    // Request that `desc` be co-located on the device where `op`
    // is placed.
    //
    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)
  4. tensorflow/c/c_api.cc

      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);
    }
    
    void TF_AddControlInput(TF_OperationDescription* desc, TF_Operation* input) {
      desc->node_builder.ControlInput(&input->node);
    }
    
    void TF_ColocateWith(TF_OperationDescription* desc, TF_Operation* op) {
    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)
Back to top