Searched refs:nested_while (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_alias_analysis_test.cc | 759 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
781 analysis.GetUniqueBufferAt(nested_while, /*index=*/{0})); in TEST_F()
788 analysis.GetUniqueBufferAt(nested_while, /*index=*/{1})); in TEST_F()
|
| H A D | hlo_dataflow_analysis_test.cc | 705 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P() local
733 EXPECT_FALSE(analysis.ValueIsDefinedAt(nested_while, /*index=*/{0})); in TEST_P()
737 EXPECT_TRUE(analysis.ValueIsDefinedAt(nested_while, /*index=*/{1})); in TEST_P()
739 analysis.GetValueDefinedAt(nested_while, /*index=*/{1}).is_phi()); in TEST_P()
753 EXPECT_THAT(HloValuesAt(nested_while, /*index=*/{0}), in TEST_P()
755 EXPECT_THAT(HloValuesAt(nested_while, /*index=*/{1}), in TEST_P()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| H A D | convert_control_to_data_outputs.mlir | 238 // CHECK-LABEL: func @nested_while
240 func.func @nested_while(%arg0: !tf_res, %arg1: tensor<f32>) {
|
| /aosp_15_r20/external/mesa3d/src/nouveau/mme/tests/ |
| H A D | mme_tu104_sim_hw_test.cpp | 1419 TEST_F(mme_tu104_sim_test, nested_while)
|