Home
last modified time | relevance | path

Searched defs:sub_builder (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dmap_test.cc473 auto sub_builder = builder.CreateSubBuilder("ErrorAdd"); in TEST_F() local
536 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F() local
563 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F() local
589 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dfusion_merger_test.cc337 HloComputation::Builder sub_builder("subcomp"); in TEST_F() local
H A Dmulti_output_fusion_test.cc740 HloComputation::Builder sub_builder("subcomp"); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Darithmetic.cc117 auto sub_builder = outer_builder->CreateSubBuilder("minmax_func"); in CreateMinMaxComputation() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbfloat16_propagation_test.cc143 auto sub_builder = HloComputation::Builder("max"); in TEST_F() local
H A Dhlo_dataflow_analysis_test.cc2831 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); in TEST_F() local
H A Dbuffer_assignment_test.cc1638 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder_test.cc414 auto sub_builder = b.CreateSubBuilder("add"); in TEST_F() local
H A Dxla_builder.cc3829 auto sub_builder = std::make_unique<XlaBuilder>(computation_name); in CreateSubBuilder() local