Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TFE_OpGetFlatInput (0.1 sec)

  1. tensorflow/c/eager/c_api.h

    //
    // Does not use the operation's definition to determine how many inputs should
    // be attached. It is intended for use with TFE_OpGetFlatInput to inspect an
    // already-finalized operation.
    //
    // Note that TFE_OpGetFlatInputCount and TFE_OpGetFlatInput operate on a flat
    // sequence of inputs, unlike TFE_OpGetInputLength (for getting the length of a
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device.cc

      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;
        const char* tensor_handle_device =
            TFE_TensorHandleDeviceName(input, 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)
  3. tensorflow/c/eager/c_api.cc

           static_cast<size_t>(num_inputs)});
    }
    
    extern int TFE_OpGetFlatInputCount(const TFE_Op* op, TF_Status* status) {
      return tensorflow::unwrap(op)->GetInputs().size();
    }
    
    extern TFE_TensorHandle* TFE_OpGetFlatInput(const TFE_Op* op, int index,
                                                TF_Status* status) {
      return tensorflow::wrap(tensorflow::unwrap(op)->GetInputs()[index]);
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top