Home
last modified time | relevance | path

Searched defs:module_ids (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dxla_debug_info_manager_test.cc45 absl::flat_hash_set<ModuleIdentifier> module_ids; in StopTracing() local
53 absl::flat_hash_set<ModuleIdentifier> module_ids; in GetModuleIds() local
H A Dhlo_module_group_test.cc194 std::vector<int64_t> module_ids; in TEST_F() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dprofiler_python.cpp1055 auto& module_ids = module_ids_[m->cls_]; in operator ()() local