Searched refs:builder_fn (Results 1 – 2 of 2) sorted by relevance
202 def builder_fn(): function211 xla_test.Benchmark(self, builder_fn, use_xla_jit=use_xla_jit, device='cpu')
267 builder_fn, argument299 name, fetches = builder_fn()