Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFE_OpGetFlatInputCount (0.34 sec)

  2. github.com/tensorflow/tensorflow

    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)
Back to top