Searched refs:CreateSubBuilder (Results 1 – 15 of 15) sorted by relevance
104 auto b = builder.CreateSubBuilder("convert"); in TEST_F()116 auto b = builder.CreateSubBuilder("convert"); in TEST_F()
472 auto sub_builder = builder.CreateSubBuilder("ErrorAdd"); in TEST_F()510 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F()537 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F()563 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F()
394 auto b = builder.CreateSubBuilder("convert"); in TEST_F()406 auto b = builder.CreateSubBuilder("convert"); in TEST_F()
195 auto b = builder.CreateSubBuilder("sum_with_rng"); in XLA_TEST_F()
245 auto sb = builder.CreateSubBuilder(absl::StrCat("branch_", i)); in XLA_TEST_P()422 auto sb = builder.CreateSubBuilder(absl::StrCat("branch_", i)); in XLA_TEST_P()
335 auto b = builder_.CreateSubBuilder("unusual"); in XLA_TEST_P()
50 builder->CreateSubBuilder(absl::StrCat(name, "_condition")); in WhileLoopHelper()63 builder->CreateSubBuilder(absl::StrCat(name, "_body")); in WhileLoopHelper()
36 b = builder->CreateSubBuilder(name); in CreateScalarComputation()38 b = builder->CreateSubBuilder( in CreateScalarComputation()118 auto sub_builder = outer_builder->CreateSubBuilder("minmax_func"); in CreateMinMaxComputation()
96 auto b = builder->CreateSubBuilder(name); in CreateScalarComparisonComputation()
195 builder->CreateSubBuilder("InvertDiagCond"); in InvertDiagonalBlocks()205 builder->CreateSubBuilder("InvertDiagBody"); in InvertDiagonalBlocks()
201 ctx->builder()->CreateSubBuilder("cond_wrapper"); in BuildWrappedCond()217 ctx->builder()->CreateSubBuilder("body_wrapper"); in BuildWrappedBody()
591 auto rb = b->CreateSubBuilder("GreaterOrEqOf_ByFirst16Bits"); in Compile()
930 builder_up = module_builder_.CreateSubBuilder(f.getName().str()); in RunOnFunction()1023 module_builder_.CreateSubBuilder(absl::StrCat("region_", region_id_++)); in LowerRegionAsComputation()
2099 auto b = CreateSubBuilder("sum"); in CrossReplicaSum()2683 std::unique_ptr<XlaBuilder> XlaBuilder::CreateSubBuilder( in CreateSubBuilder() function in xla::XlaBuilder
227 std::unique_ptr<XlaBuilder> CreateSubBuilder(const string& computation_name);