| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_alias_analysis_test.cc | 366 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
551 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
644 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
815 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
|
| H A 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
|
| H A D | copy_insertion_test.cc | 1359 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
1634 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
1696 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
1773 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
1847 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
|
| H A D | hlo_dataflow_analysis_test.cc | 400 auto body_builder = HloComputation::Builder("body"); in TEST_P() local
503 auto body_builder = HloComputation::Builder("body"); in TEST_P() local
786 auto body_builder = HloComputation::Builder("body"); in TEST_P() local
839 auto body_builder = HloComputation::Builder("body"); in TEST_P() local
1366 auto body_builder = HloComputation::Builder(TestName()); in TEST_P() local
|
| H A D | heap_simulator_test.cc | 62 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
154 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
905 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
|
| H A D | cholesky_expander.cc | 77 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in CholeskyUnblocked()
|
| H A D | hlo_live_range_test.cc | 282 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
|
| H A D | hlo_dce_test.cc | 269 auto body_builder = HloComputation::Builder(TestName() + "-body"); in TEST_F() local
|
| H A D | eigh_expander.cc | 324 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in Sweeps()
|
| H A D | memory_space_assignment_test.cc | 794 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_P() local
2603 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_P() local
2925 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_P() local
3016 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_P() local
|
| H A D | dynamic_dimension_inference_test.cc | 749 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
|
| H A D | buffer_assignment_test.cc | 814 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | loops.cc | 62 std::unique_ptr<XlaBuilder> body_builder = in WhileLoopHelper() local
94 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in ForEachIndex()
|
| H A D | svd.cc | 277 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in HouseHolderBidiagonalization()
626 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in WhileLoopFn()
|
| H A D | math.cc | 1664 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in LentzThompsonBarnettAlgorithm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | outfeed_in_nested_computation_test.cc | 52 XlaBuilder body_builder("loop_body"); in XLA_TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/graph_to_func/ |
| H A D | graph_to_func.cc | 82 OpBuilder body_builder = OpBuilder::atBlockEnd(func_op.getBody()); in GraphToFunc() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
| H A D | functiondef_import.cc | 418 OpBuilder body_builder = OpBuilder::atBlockEnd(body); in ImportGenericFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | import_model.cc | 3364 mlir::OpBuilder body_builder(&func.getBody()); in CreateSavedModelIR() local
|