| /external/tensorflow/tensorflow/lite/ |
| D | tensorflow_profiler_logger.cc | 123 tensorflow::profiler::TraceMe* trace_me = in OnTfLiteSubgraphInvoke() local
133 void OnTfLiteSubgraphInvokeEnd(tensorflow::profiler::TraceMe* trace_me) { in OnTfLiteSubgraphInvokeEnd()
146 tensorflow::profiler::TraceMe* trace_me = new tensorflow::profiler::TraceMe( in OnTfLiteOpInvoke() local
159 void OnTfLiteOpInvokeEnd(tensorflow::profiler::TraceMe* trace_me) { in OnTfLiteOpInvokeEnd()
|
| D | tensorflow_profiler_logger_shim.cc | 42 tensorflow::profiler::TraceMe* trace_me) {} in OnTfLiteSubgraphInvokeEnd()
54 tensorflow::profiler::TraceMe* trace_me) {} in OnTfLiteOpInvokeEnd()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_reshard_variables_op_util.cc | 82 profiler::TraceMe trace_me("TPUReshardVariablesOpKernel::LookupProto", in GetComputationCacheEntry() local
102 profiler::TraceMe trace_me("BuildComputationInputs", /*level=*/2); in BuildInputBuffers() local
212 profiler::TraceMe trace_me("PerformCompaction", /*level=*/2); in PerformCompaction() local
230 profiler::TraceMe trace_me("UpdateOutputVariables", /*level=*/2); in UpdateOutputVariables() local
|
| D | tpu_execute_op.cc | 73 profiler::TraceMe trace_me("TpuExecuteOp::LookupProto", /*level=*/2); in GetComputationCacheEntry() local
199 profiler::TraceMe trace_me("BuildComputationInputs", /*level=*/2); in BuildComputationInputs() local
401 profiler::TraceMe trace_me("AllocateOutputTensors", /*level=*/2); in AllocateOutputTensors() local
626 profiler::TraceMe trace_me( in DoWork() local
|
| D | tpu_reshard_variables_op.cc | 133 profiler::TraceMe trace_me( in DoTpuExecute() local
|
| D | tpu_functional_ops.cc | 1272 profiler::TraceMe trace_me( in ComputeAsync() local
1361 profiler::TraceMe trace_me("TPUPartitionedCallOp-GetTpuCoreOrdinal"); in GetTpuCoreOrdinal() local
1432 profiler::TraceMe trace_me("TPUPartitionedCallOp-InitializeVarOnTPU"); in InitializeVarOnTPU() local
1605 profiler::TraceMe trace_me( in InitializeShardedVarOnTPU() local
2637 profiler::TraceMe trace_me("TPUPartitionedCallOp-ExecuteRemote"); in ExecuteRemoteFunction() local
2663 profiler::TraceMe trace_me("TPUPartitionedCallOp-ExecuteLocal"); in ExecuteLocalFunction() local
2681 profiler::TraceMe trace_me("TPUPartitionedCallOp-ExecuteFunctions"); in ExecuteFunctions() local
|
| D | infeed_ops.cc | 78 profiler::TraceMe trace_me("TransposeTensor", /*level=*/2); in TransposeTensor() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | functional_ops.cc | 175 profiler::TraceMe trace_me("IfOp"); in Start() local
305 profiler::TraceMe trace_me("CaseOp"); in Start() local
535 profiler::TraceMe trace_me("WhileOp-EvalCond"); in EvalCond() local
571 profiler::TraceMe trace_me("WhileOp-StartBody"); in StartBody() local
626 profiler::TraceMe trace_me("WhileOp-EvalCond"); in DoComputeSync() local
647 profiler::TraceMe trace_me("WhileOp-StartBody"); in DoComputeSync() local
827 profiler::TraceMe trace_me("ForOp"); in RunNext() local
|
| D | function_ops.cc | 251 profiler::TraceMe trace_me("SymbolicGradientOp"); in ComputeAsync() local
393 profiler::TraceMe trace_me( in ComputeAsync() local
|
| D | partitioned_function_ops.cc | 261 profiler::TraceMe trace_me("PartitionedCallOp"); in RunFunction() local
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 64 profiler::TraceMe trace_me("XRTExecuteOp::LookupProto", /*level=*/2); in GetComputationCacheEntry() local
195 profiler::TraceMe trace_me("RunExecutable", /*level=*/2); in RunExecutable() local
284 profiler::TraceMe trace_me( in DoWork() local
405 profiler::TraceMe trace_me( in DoWork() local
|
| /external/pytorch/test/jit/ |
| D | test_python_ir.py | 23 def trace_me(arg): function
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_memory_manager.cc | 104 profiler::TraceMe trace_me("XRTMemoryManager::CompactAllocations", in CompactAllocations() local
154 profiler::TraceMe trace_me("XRTMemoryManager::TryFreeMemory", /*level=*/2); in TryFreeMemory() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_jitrt_kernels.cc | 377 TraceMe trace_me([&] { in CompileImpl() local
427 TraceMe trace_me([&] { in CompileImpl() local
653 TraceMe trace_me([&] { in ExecuteImpl() local
|
| D | tf_cpurt_kernels.cc | 270 profiler::TraceMe trace_me([&] { in ExecuteImpl() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 491 tensorflow::profiler::TraceMe trace_me([&]() -> std::string { in KernelFallbackExecuteOpInternal() local
680 tensorflow::profiler::TraceMe trace_me("tfrt_fallback_async.get_resource"); in FallbackGetResource() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_kernels.cc | 981 tensorflow::profiler::TraceMe trace_me( in CoreRTTensorHandleToFallbackTensor() local
1024 tensorflow::profiler::TraceMe trace_me( in FallbackTensorToCoreRTTensorHandle() local
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 430 profiler::TraceMe trace_me([padded_batch_size, padding_amount]() { in ConcatInputTensors() local
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1149 tensorflow::profiler::TraceMe trace_me("PjRtStreamExecutorBuffer::Release"); in Release() local
|