Searched refs:nested_builder (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 210 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange args) { in matchAndRewrite() argument
219 nested_builder.create<linalg::YieldOp>(loc, op_result); in matchAndRewrite()
395 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange args) { in matchAndRewrite() argument
396 nested_builder.create<linalg::YieldOp>(loc, *args.begin()); in matchAndRewrite()
524 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange args) { in matchAndRewrite() argument
525 nested_builder.create<linalg::YieldOp>(loc, *args.begin()); in matchAndRewrite()
567 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange args) { in matchAndRewrite() argument
568 nested_builder.create<linalg::YieldOp>(loc, val); in matchAndRewrite()
578 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange args) { in matchAndRewrite() argument
579 nested_builder.create<linalg::YieldOp>(loc, *args.begin()); in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 671 auto nested_builder = HloComputation::Builder("mul"); in XLA_TEST_F() local
673 auto x = nested_builder.AddInstruction( in XLA_TEST_F()
675 auto y = nested_builder.AddInstruction( in XLA_TEST_F()
677 nested_builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
681 hlo_module->AddEmbeddedComputation(nested_builder.Build()); in XLA_TEST_F()
|
| /external/flatbuffers/tests/ |
| D | test.cpp | 177 flatbuffers::FlatBufferBuilder nested_builder; in CreateFlatBufferTest() local
178 auto nmloc = CreateMonster(nested_builder, nullptr, 0, 0, in CreateFlatBufferTest()
179 nested_builder.CreateString("NestedMonster")); in CreateFlatBufferTest()
180 FinishMonsterBuffer(nested_builder, nmloc); in CreateFlatBufferTest()
185 builder.ForceVectorAlignment(nested_builder.GetSize(), sizeof(uint8_t), in CreateFlatBufferTest()
186 nested_builder.GetBufferMinAlignment()); in CreateFlatBufferTest()
191 nested_builder.GetBufferPointer(), nested_builder.GetSize()); in CreateFlatBufferTest()
|