- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TFE_OpAddInput (0.07 sec)
-
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) -
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)