Searched refs:MakeSortHlo (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.h | 136 StatusOr<HloInstruction*> MakeSortHlo(
|
| D | bfloat16_normalization_test.cc | 288 MakeSortHlo(ShapeUtil::MakeTupleShape({bf16_shape, s32_shape}), in TEST_F()
317 MakeSortHlo(ShapeUtil::MakeTupleShape({bf16_shape, f32_shape}), in TEST_F()
|
| D | hlo_creation_utils.cc | 288 StatusOr<HloInstruction*> MakeSortHlo( in MakeSortHlo() function
|
| D | tuple_points_to_analysis_test.cc | 1075 auto* sort, MakeSortHlo(keys_shape, {keys}, 0, /*is_stable=*/false, in TEST_F()
1097 MakeSortHlo(ShapeUtil::MakeTupleShape({keys_shape, values_shape}), in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 2454 auto* sort, MakeSortHlo(keys_shape, {keys}, -1, /*is_stable=*/false, in TEST_F()
2476 MakeSortHlo(ShapeUtil::MakeTupleShape({keys_shape, values_shape}), in TEST_F()
|
| D | algebraic_simplifier_test.cc | 2807 TF_ASSERT_OK(MakeSortHlo(keys_shape, {keys}, 0, /*is_stable=*/false, &builder, in TEST_F()
2828 TF_ASSERT_OK(MakeSortHlo(ShapeUtil::MakeTupleShape( in TEST_F()
|