(batch_size, value)
| 33 | |
| 34 | |
| 35 | def get_data(batch_size, value): |
| 36 | pipe = get_dali_pipe(batch_size=batch_size, device_id=None, num_threads=1, value=value) |
| 37 | daliop = dali_tf.DALIIterator() |
| 38 | out = [] |
| 39 | with tf.device("/cpu"): |
| 40 | data = daliop( |
| 41 | pipeline=pipe, |
| 42 | shapes=[batch_size], |
| 43 | dtypes=[tf.int32], |
| 44 | device_id=None, |
| 45 | ) |
| 46 | out.append(data) |
| 47 | return [out] |
| 48 | |
| 49 | |
| 50 | def test_dali_tf_op_cpu_only(): |
no test coverage detected