def test_audio_decoder_cpu():
pipe = Pipeline(batch_size=batch_size, num_threads=4, device_id=None)
input, _ = fn.file_reader(file_root=audio_dir, shard_id=0, num_shards=1)
decoded, _ = fn.audio_decoder(input)
pipe.set_outputs(decoded)
pipe.build()
for _ in range(3):
pipe.run()
def audio_decoder_pipe(max_batch_size, input_data, device):
pipe = Pipeline(batch_size=max_batch_size, num_threads=4, device_id=0)
encoded = fn.external_source(source=input_data,
cycle=False,
device='cpu')
decoded, _ = fn.audio_decoder(encoded,
downmix=True,
sample_rate=12345,
device=device)
pipe.set_outputs(decoded)
return pipe
