Searched refs:d_builder (Results 1 – 2 of 2) sorted by relevance
334 NodeBuilder d_builder("D", "SymbolicGradient", in TEST() local338 d_builder.Attr("f", d_name_attr); in TEST()339 d_builder.Attr("Tin", {DT_FLOAT, DT_FLOAT}); in TEST()340 d_builder.Attr("Tout", {DT_FLOAT}); in TEST()341 d_builder.Input({c, c}); in TEST()342 builder.opts().FinalizeBuilder(&d_builder); in TEST()
1430 auto d_builder = HloComputation::Builder(TestName() + "_d"); in TEST_F() local1431 auto d_param = d_builder.AddInstruction( in TEST_F()1433 auto d_computation = d_builder.Build(); in TEST_F()