Skip to content

Instantly share code, notes, and snippets.

@ivelin

ivelin/pcap_input.cc

Created June 19, 2019 04:43
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ivelin/e5cd70cf9a356711e70d0086e07c10b7 to your computer and use it in GitHub Desktop.
Save ivelin/e5cd70cf9a356711e70d0086e07c10b7 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pcap_input.cc
REGISTER_UNARY_VARIANT_DECODE_FUNCTION(PcapInput, "tensorflow::data::PcapInput");
REGISTER_KERNEL_BUILDER(Name("PcapInput").Device(DEVICE_CPU),
FileInputOp<PcapInput>);
REGISTER_KERNEL_BUILDER(Name("PcapDataset").Device(DEVICE_CPU),
FileInputDatasetOp<PcapInput, PcapInputStream>);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment