Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFE_OpGetFlatInput (0.17 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/c/eager/custom_device_testutil.cc 

      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;
    const char* input_device = TFE_TensorHandleDeviceName(input, s);
    if (TF_GetCode(s) != TF_OK) return;
    if (dev->device_name == input_device) {
    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