Searched refs:report_input_shapes (Results 1 – 14 of 14) sorted by relevance
| /external/pytorch/torch/csrc/profiler/standalone/ |
| D | nvtx_observer.cpp | 126 template <bool report_input_shapes>
134 report_input_shapes ? torch::profiler::impl::inputSizes(fn, true) in enterNVTX()
137 report_input_shapes in enterNVTX()
161 state_ptr->config().report_input_shapes in pushNVTXCallbacks()
168 .needsInputs(config.report_input_shapes) in pushNVTXCallbacks()
169 .needsOutputs(config.report_input_shapes) in pushNVTXCallbacks()
|
| D | itt_observer.cpp | 34 template <bool report_input_shapes>
58 state_ptr->config().report_input_shapes in pushITTCallbacks()
64 .needsInputs(config.report_input_shapes) in pushITTCallbacks()
|
| /external/pytorch/torch/csrc/profiler/orchestration/ |
| D | observer.cpp | 36 bool report_input_shapes, in ProfilerConfig() argument
44 report_input_shapes{report_input_shapes},
71 eventIValueList.emplace_back(report_input_shapes); in toIValue()
|
| D | observer.h | 93 bool report_input_shapes = false,
105 bool report_input_shapes; member
|
| /external/pytorch/torch/csrc/profiler/ |
| D | kineto_client_interface.cpp | 28 bool report_input_shapes = false, in prepare() argument
33 reportInputShapes_ = report_input_shapes; in prepare()
|
| D | collection.cpp | 336 if (config_.report_input_shapes) { in begin_op()
1516 if (config_.report_input_shapes && config_.profile_memory) { in getRecords()
|
| /external/pytorch/torch/csrc/jit/mobile/ |
| D | profiler_edge.cpp | 15 const bool report_input_shapes, in KinetoEdgeCPUProfiler() argument
34 report_input_shapes, in KinetoEdgeCPUProfiler()
|
| D | profiler_edge.h | 55 const bool report_input_shapes = false,
|
| /external/pytorch/torch/csrc/distributed/rpc/ |
| D | request_callback_no_python.cpp | 412 profilingConfig.report_input_shapes, in processRunWithProfilingReq()
421 profilingConfig.report_input_shapes, in processRunWithProfilingReq()
|
| /external/pytorch/torch/csrc/autograd/ |
| D | profiler_legacy.cpp | 382 if (state_ptr->config().report_input_shapes) { in pushProfilingCallbacksLegacy()
405 .needsInputs(registration_state_ptr->config().report_input_shapes) in pushProfilingCallbacksLegacy()
|
| D | profiler_kineto.cpp | 523 .needsInputs(registration_state_ptr->config().report_input_shapes) in pushProfilingCallbacks()
|
| D | python_function.cpp | 878 torch::autograd::profiler::getProfilerConfig().report_input_shapes; in unpack_input()
|
| /external/pytorch/torch/_C/ |
| D | _profiler.pyi | 69 report_input_shapes: bool,
|
| /external/pytorch/torch/csrc/profiler/python/ |
| D | init.cpp | 227 torch::autograd::profiler::getProfilerConfig().report_input_shapes; in RecordFunctionFast_enter()
|