pytorch/binaries
Ilia Cherniavskii 800d5617c0 Recording of TorchScript functions (#34710)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34710

Extending RecordFunction API to support new recording scopes (such as TorchScript functions), as well as giving more flexibility to set sampling rate.

Test Plan: unit test (test_misc.cpp/testRecordFunction)

Reviewed By: gdankel, dzhulgakov

Differential Revision: D20158523

fbshipit-source-id: a9e0819d21cc06f4952d92d43246587c36137582
2020-03-31 00:33:23 -07:00
..
bench_gen
at_launch_benchmark.cc
benchmark_args.h
benchmark_helper.cc
benchmark_helper.h [aibench] added support for measuring memory on AI Bench for Caffe2 Models (#35036) 2020-03-25 01:58:18 -07:00
caffe2_benchmark.cc
CMakeLists.txt Recording of TorchScript functions (#34710) 2020-03-31 00:33:23 -07:00
convert_and_benchmark.cc
convert_caffe_image_db.cc
convert_db.cc
convert_encoded_to_raw_leveldb.cc
convert_image_to_tensor.cc
core_overhead_benchmark.cc
core_overhead_benchmark_gpu.cc
db_throughput.cc
dump_operator_names.cc [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
inspect_gpu.cc
intra_inter_benchmark.cc
lite_interpreter_model_load.cc [pytorch][mobile] fixed AutoGradMode/AutoNonVariableTypeMode uses for mobile callsites 2020-03-18 17:19:35 -07:00
make_cifar_db.cc
make_image_db.cc
make_mnist_db.cc Enable -Werror=format compile errors on torch exception types (#34019) 2020-03-02 13:25:39 -08:00
parallel_info.cc
predictor_verifier.cc
print_core_object_sizes_gpu.cc
print_registered_core_operators.cc
record_function_benchmark.cc Recording of TorchScript functions (#34710) 2020-03-31 00:33:23 -07:00
run_plan.cc
run_plan_mpi.cc
speed_benchmark.cc
speed_benchmark_torch.cc [On-device Benchmark] speed_benchmark_torch switch to log latency from dataset level to row level (#34598) 2020-03-11 13:51:42 -07:00
split_db.cc
tsv_2_proto.cc
tutorial_blob.cc
zmq_feeder.cc