| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_profile_printer.cc | 40 return counters[instruction_info.profile_index()] != 0; in PrintHloProfile()
52 counters[computation_info.profile_index()], clock_rate_ghz); in PrintHloProfile()
59 counters[instruction_info.profile_index()], in PrintHloProfile()
|
| D | hlo_profile_printer_data.proto | 39 int64 profile_index = 8; field
48 int64 profile_index = 2; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_execution_profiler.h | 59 void FinishHloComputation(absl::optional<size_t> profile_index);
67 void FinishHloInstruction(size_t profile_index);
75 absl::optional<int64> profile_index);
|
| D | thunk.h | 75 absl::optional<int64> profile_index; member
84 profile_index_(thunk_info.profile_index), in Thunk()
126 absl::optional<int64> profile_index() const { return profile_index_; } in profile_index() function
|
| D | memset_thunk.cc | 28 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
37 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | copy_thunk.cc | 36 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
55 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | cudnn_batchnorm_thunk.cc | 54 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
103 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
151 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | hlo_execution_profiler.cc | 93 absl::optional<size_t> profile_index) { in FinishHloComputation() argument
96 *profile_index, in FinishHloComputation()
|
| D | replica_id_thunk.cc | 23 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | gemm_thunk.cc | 68 implements_whole_instruction_, profile_index(), in ExecuteOnStream()
175 absl::optional<int64> profile_index, in RunGemm() argument
249 implements_whole_instruction ? profile_index : -1) in RunGemm()
|
| D | sequential_thunk.cc | 41 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | for_thunk.cc | 47 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | convolution_thunk.cc | 58 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | gemm_thunk.h | 83 absl::optional<int64> profile_index,
|
| D | infeed_thunk.cc | 40 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | while_thunk.cc | 62 auto op_profiler = profiler.MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | nccl_collective_thunk.cc | 68 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | outfeed_thunk.cc | 40 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | conditional_thunk.cc | 74 auto op_profiler = profiler.MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | kernel_thunk.cc | 117 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | collective_permute_thunk.cc | 232 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| D | gpu_compiler.cc | 942 size_t profile_index = 0; in RunBackend() local
944 profile_index = in RunBackend()
954 std::move(module), profile_index, std::move(profile_printer), in RunBackend()
|
| D | fft_thunk.cc | 130 params.profiler->MakeScopedInstructionProfiler(profile_index()); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.cc | 105 Status GetTrtBindingIndex(const char* tensor_name, int profile_index, in GetTrtBindingIndex() argument
130 *binding_index = *binding_index + profile_index * bindings_per_profile; in GetTrtBindingIndex()
|
| D | trt_engine_utils.h | 128 Status GetTrtBindingIndex(const char* tensor_name, int profile_index,
|