- Sort Score
- Num 10 results
- Language All
Results 1 - 1 of 1 for TFE_OpGetFlatInput (0.06 seconds)
-
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) {
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat Dec 20 23:30:21 GMT 2025 - 8.4K bytes - Click Count (0)