Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFE_OpAddInput (0.44 sec)

  1. tensorflow/c/eager/custom_device_testutil.cc

          LoggedTensor* t = reinterpret_cast<LoggedTensor*>(
              TFE_TensorHandleDevicePointer(input, s));
          if (TF_GetCode(s) != TF_OK) return;
          TFE_OpAddInput(op, t->tensor, s);
        } else {
          TFE_OpAddInput(op, input, s);
        }
        if (TF_GetCode(s) != TF_OK) return;
      }
      std::vector<TFE_TensorHandle*> op_outputs(*num_outputs);
      TFE_Execute(op, op_outputs.data(), num_outputs, s);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 03 20:47:31 GMT 2021
    - 8.3K bytes
    - Viewed (0)
Back to top