Searched defs:root_tuple (Results 1 – 3 of 3) sorted by relevance
44 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() local64 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() local83 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() local104 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() local124 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() local144 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() local166 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() local188 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() local211 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() local
437 auto root_tuple = body->root_instruction(); in GroupHlosForConcat() local
222 const HloInstruction* root_tuple, in AreTransitiveUsesEffectivelyElementwise()