Searched defs:embedded_builder (Results 1 – 9 of 9) sorted by relevance
85 auto embedded_builder = HloComputation::Builder("add"); in TEST_F() local
118 auto embedded_builder = HloComputation::Builder("add"); in TEST_F() local
82 auto embedded_builder = HloComputation::Builder("add"); in TEST_F() local
57 auto embedded_builder = HloComputation::Builder("add"); in GetAdd() local68 auto embedded_builder = HloComputation::Builder("add"); in GetAddTuple() local88 auto embedded_builder = HloComputation::Builder("ge"); in GetGe() local
336 auto embedded_builder = HloComputation::Builder("f32+1"); in TEST_F() local374 auto embedded_builder = HloComputation::Builder("f32+f32"); in TEST_F() local
99 auto embedded_builder = HloComputation::Builder("add"); in GetScalarAddComputation() local
399 auto embedded_builder = HloComputation::Builder("add"); in RewriteDynamicReshapeSplitInput() local
1651 auto embedded_builder = HloComputation::Builder(TestName() + "_embedded"); in TEST_P() local
159 auto embedded_builder = HloComputation::Builder("f32+f32"); in TEST_F() local