Searched refs:reduce_subcomp (Results 1 – 1 of 1) sorted by relevance
339 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F() local348 /*dimensions_to_reduce=*/{0}, reduce_subcomp)); in TEST_F()380 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F() local389 /*dimensions_to_reduce=*/{0}, reduce_subcomp)); in TEST_F()399 /*dimensions_to_reduce=*/{0}, reduce_subcomp)); in TEST_F()