Searched refs:TFE_OpGetFlatInputCount (Results 1 – 5 of 5) sorted by relevance
114 int num_inputs = TFE_OpGetFlatInputCount(original_op, s); in LoggingDeviceExecute()
284 TF_CAPI_EXPORT extern int TFE_OpGetFlatInputCount(const TFE_Op* op,
684 extern int TFE_OpGetFlatInputCount(const TFE_Op* op, TF_Status* status) { in TFE_OpGetFlatInputCount() function
1763 int num_inputs = TFE_OpGetFlatInputCount(other, status); in CloneOp()
297 int num_inputs = TFE_OpGetFlatInputCount(original_op, status); in ParallelDeviceExecute()