MCPcopy Create free account
hub / github.com/NVIDIA/DALI / test_tfrecord_reader_alias

Function test_tfrecord_reader_alias

dali/test/python/reader/test_index.py:238–243  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

236
237
238def test_tfrecord_reader_alias():
239 tfrecord = os.path.join(get_dali_extra_path(), "db", "tfrecord", "train")
240 tfrecord_idx = os.path.join(get_dali_extra_path(), "db", "tfrecord", "train.idx")
241 new_pipe = tfrecord_pipe(fn.readers.tfrecord, tfrecord, tfrecord_idx)
242 legacy_pipe = tfrecord_pipe(fn.tfrecord_reader, tfrecord, tfrecord_idx)
243 compare_pipelines(new_pipe, legacy_pipe, batch_size_alias_test, 50)
244
245
246@pipeline_def(batch_size=batch_size_alias_test, device_id=0, num_threads=4)

Callers

nothing calls this directly

Calls 3

get_dali_extra_pathFunction · 0.90
compare_pipelinesFunction · 0.90
tfrecord_pipeFunction · 0.85

Tested by

no test coverage detected