Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFE_OpAddAttrs (0.3 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/c/eager/custom_device_testutil.cc 

      if (TF_GetCode(s) != TF_OK) return;
  const TFE_OpAttrs* attributes = TFE_OpGetAttrs(original_op);

  TFE_Op* op(TFE_NewOp(context, operation_name, s));
  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;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Mar 03 20:47:31 UTC 2021
    - 8.3K bytes
    - Viewed (0)
Back to top