Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TFE_OpAddInput (0.07 sec)

  1. tensorflow/c/eager/c_api_remote_test_util.cc

        matmul = TFE_NewOp(ctx, "MatMulFunction", status);
        ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
        TFE_OpAddInput(matmul, h0_task0, status);
        ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
        TFE_OpAddInput(matmul, has_packed_input ? packed_handle : h1_task2, status);
        ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
      } else {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Dec 11 22:56:03 UTC 2020
    - 9.1K bytes
    - Viewed (0)
  2. 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);
    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