| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_affine.cc | 35 function_ref<void(OpBuilder&, Location, ValueRange)> body_builder) { in BuildBoundedAffineLoopNest()
75 auto body_builder = [&](OpBuilder& builder, Location loc, ValueRange ivs) { in matchAndRewrite() local
118 auto body_builder = [&](OpBuilder& builder, Location loc, in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | loops.cc | 62 std::unique_ptr<XlaBuilder> body_builder = in WhileLoopHelper() local
94 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in ForEachIndex()
|
| D | svd.cc | 276 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in HouseHolderBidiagonalization()
611 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in WhileLoopFn()
|
| D | self_adjoint_eig.cc | 242 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in WhileLoopFn()
|
| D | math.cc | 1666 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in LentzThompsonBarnettAlgorithm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_alias_analysis_test.cc | 333 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
519 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
614 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
785 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
935 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
1212 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
|
| D | hlo_ordering_test.cc | 131 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
214 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
|
| D | cholesky_expander.cc | 79 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in CholeskyUnblocked()
|
| D | hlo_dataflow_analysis_test.cc | 393 auto body_builder = HloComputation::Builder("body"); in TEST_P() local
496 auto body_builder = HloComputation::Builder("body"); in TEST_P() local
779 auto body_builder = HloComputation::Builder("body"); in TEST_P() local
832 auto body_builder = HloComputation::Builder("body"); in TEST_P() local
1079 auto body_builder = HloComputation::Builder("body"); in TEST_P() local
1439 auto body_builder = HloComputation::Builder(TestName()); in TEST_P() local
|
| D | hlo_live_range_test.cc | 282 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
|
| D | copy_insertion_test.cc | 1513 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
1788 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
1850 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
1927 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
2001 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
|
| D | hlo_dce_test.cc | 271 auto body_builder = HloComputation::Builder(TestName() + "-body"); in TEST_F() local
|
| D | heap_simulator_test.cc | 63 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
155 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
905 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
|
| D | memory_space_assignment_test.cc | 759 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_P() local
2642 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_P() local
2964 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_P() local
3055 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_P() local
|
| D | dynamic_dimension_inference_test.cc | 742 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
|
| D | buffer_assignment_test.cc | 795 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | outfeed_in_nested_computation_test.cc | 52 XlaBuilder body_builder("loop_body"); in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 3124 mlir::OpBuilder body_builder(&func.getBody()); in CreateSavedModelIR() local
|