Searched refs:map_builder (Results 1 – 3 of 3) sorted by relevance
322 auto* map_builder = ebuilder->getLayerBuilder(ParseDefault((*jv)["k"], -1)); in GetDisplacementSource() local323 if (map_builder) { in GetDisplacementSource()324 return std::make_tuple(map_builder->contentTree(), map_builder->size()); in GetDisplacementSource()
1084 HloComputation::Builder map_builder("id"); in TEST_F() local1085 map_builder.AddInstruction( in TEST_F()1087 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F()
1375 auto map_builder = HloComputation::Builder(TestName() + "_map"); in TEST_F() local1376 auto map_param = map_builder.AddInstruction( in TEST_F()1378 auto map_root = map_builder.AddInstruction( in TEST_F()1380 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F()