Searched refs:FunctionVisitor (Results 1 – 10 of 10) sorted by relevance
591 class FunctionVisitor(transformer.Base): class595 super(FunctionVisitor, self).__init__(source_info)626 visitor = FunctionVisitor(source_info, graphs, resolver)
229 FunctionVisitor visit_fn([this, &foldable_dots, &foldable_convolutions]( in Run()
127 FunctionVisitor function_visitor(visitor); in Build()390 FunctionVisitor function_visitor(visitor); in RecordInstructions()
489 FunctionVisitor visitor([&sequence](HloInstruction* hlo) { in DFSMemoryScheduler()746 FunctionVisitor visitor( in Run()
348 using FunctionVisitor = FunctionVisitorBase<HloInstruction*>; variable
216 absl::make_unique<FunctionVisitor>([](HloInstruction*) { in HloEvaluator()221 absl::make_unique<FunctionVisitor>([](HloInstruction*) { in HloEvaluator()226 absl::make_unique<FunctionVisitor>([](HloInstruction*) { in HloEvaluator()
434 FunctionVisitor visitor( in TEST_F()
1022 TEST_F(HloInstructionTest, FunctionVisitor) { in TEST_F() argument1046 FunctionVisitor visitor([&visit_num, &visit_order](HloInstruction* inst) { in TEST_F()
278 class FunctionVisitor(PrototypeVisitor): class1223 FunctionVisitor(f),
311 class FunctionVisitor(PrototypeVisitor): class1463 FunctionVisitor(f),