Searched refs:profile_index (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_profile_printer.cc | 41 return counters[instruction_info.profile_index()] != 0; in PrintHloProfile()
53 counters[computation_info.profile_index()], clock_rate_ghz); in PrintHloProfile()
60 counters[instruction_info.profile_index()], in PrintHloProfile()
|
| D | hlo_profile_printer_data.proto | 40 int64 profile_index = 8; field
49 int64 profile_index = 2; field
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/common/ |
| D | utils.cc | 60 Status GetTrtBindingIndex(const char* tensor_name, int profile_index, in GetTrtBindingIndex() argument
82 *binding_index = *binding_index + profile_index * bindings_per_profile; in GetTrtBindingIndex()
86 Status GetTrtBindingIndex(int network_input_index, int profile_index, in GetTrtBindingIndex() argument
91 return GetTrtBindingIndex(input_name.c_str(), profile_index, cuda_engine, in GetTrtBindingIndex()
|
| D | utils.h | 113 Status GetTrtBindingIndex(const char* tensor_name, int profile_index,
120 Status GetTrtBindingIndex(int network_input_idx, int profile_index,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.h | 75 std::optional<int64_t> profile_index; member
84 profile_index_(thunk_info.profile_index), in Thunk()
127 std::optional<int64_t> profile_index() const { return profile_index_; } in profile_index() function
|