Searched refs:CreateSubBuilder (Results 1 – 14 of 14) 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()
187 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()
312 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()
39 b = builder->CreateSubBuilder(name); in CreateScalarComputation()41 b = builder->CreateSubBuilder( in CreateScalarComputation()
97 auto b = builder->CreateSubBuilder(name); in CreateScalarComparisonComputation()
195 builder->CreateSubBuilder("InvertDiagCond"); in InvertDiagonalBlocks()205 builder->CreateSubBuilder("InvertDiagBody"); in InvertDiagonalBlocks()
183 ctx->builder()->CreateSubBuilder("cond_wrapper"); in BuildWrappedCond()199 ctx->builder()->CreateSubBuilder("body_wrapper"); in BuildWrappedBody()
576 auto rb = b->CreateSubBuilder("GreaterOrEqOf_ByFirst16Bits"); in Compile()
1635 auto b = this->CreateSubBuilder("comparator"); in Sort()2125 auto b = CreateSubBuilder("sum"); in CrossReplicaSum()2637 std::unique_ptr<XlaBuilder> XlaBuilder::CreateSubBuilder( in CreateSubBuilder() function in xla::XlaBuilder
198 std::unique_ptr<XlaBuilder> CreateSubBuilder(const string& computation_name);