Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_dimension_inference_test.cc64 auto embedded_builder = HloComputation::Builder("add"); in GetAdd() local
75 auto embedded_builder = HloComputation::Builder("add"); in GetAddTuple() local
95 auto embedded_builder = HloComputation::Builder("ge"); in GetGe() local
H A Dhlo_instruction_test.cc337 auto embedded_builder = HloComputation::Builder("f32+1"); in TEST_F() local
375 auto embedded_builder = HloComputation::Builder("f32+f32"); in TEST_F() local
H A Ddynamic_padder_test.cc109 auto embedded_builder = HloComputation::Builder("add"); in GetScalarAddComputation() local
H A Ddynamic_padder.cc500 auto embedded_builder = HloComputation::Builder("add"); in RewriteDynamicReshapeSplitInput() local
H A Dhlo_dataflow_analysis_test.cc1579 auto embedded_builder = HloComputation::Builder(TestName() + "_embedded"); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dgpu_ldg_test.cc117 auto embedded_builder = HloComputation::Builder("add"); in TEST_F() local
H A Dparallel_reduction_test.cc123 auto embedded_builder = HloComputation::Builder("add"); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dhorizontal_input_fusion_test.cc85 auto embedded_builder = HloComputation::Builder("add"); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
H A Dcpu_fusion_test.cc158 auto embedded_builder = HloComputation::Builder("f32+f32"); in TEST_F() local