Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for TFE_OpGetFlatInputCount (0.42 sec)

  1. tensorflow/c/eager/custom_device_testutil.cc

      if (TF_GetCode(s) != TF_OK) return;
      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: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 03 20:47:31 UTC 2021
    - 8.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.h

    // 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
    // particular named input list, which may only be part of the op's inputs).
    TF_CAPI_EXPORT extern int TFE_OpGetFlatInputCount(const TFE_Op* op,
                                                      TF_Status* status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device.cc

      NamedParallelDevice* named_device =
          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);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 29 22:05:31 UTC 2023
    - 18.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api.cc

          {reinterpret_cast<tensorflow::AbstractTensorHandle**>(
               tensorflow::unwrap(inputs)),
           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,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_test.cc

      TFE_OpSetDevice(ret, device, status);
      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) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
Back to top