Searched refs:outer_param (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | copy_insertion_test.cc | 773 HloInstruction* outer_param = outer_while_body->parameter_instruction(0); in TEST_F() local
775 for (int i = 0; i < outer_param->shape().tuple_shapes_size(); ++i) { in TEST_F()
778 outer_param->shape().tuple_shapes(i), outer_param, i))); in TEST_F()
840 HloInstruction* outer_param = outer_while_body->parameter_instruction(0); in TEST_F() local
844 while_hlo->while_body(), outer_param)); in TEST_F()
899 HloInstruction* outer_param = outer_while_body->parameter_instruction(0); in TEST_F() local
901 for (int i = 0; i < outer_param->shape().tuple_shapes_size(); ++i) { in TEST_F()
904 outer_param->shape().tuple_shapes(i), outer_param, i))); in TEST_F()
|
| H A D | hlo_alias_analysis_test.cc | 749 auto outer_param = outer_builder.AddInstruction( in TEST_F() local
752 HloInstruction::CreateGetTupleElement(scalar_shape_, outer_param, 0)); in TEST_F()
756 HloInstruction::CreateGetTupleElement(scalar_shape_, outer_param, 1)); in TEST_F()
|
| H A D | hlo_dataflow_analysis_test.cc | 695 auto outer_param = outer_builder.AddInstruction( in TEST_P() local
698 HloInstruction::CreateGetTupleElement(scalar_shape_, outer_param, 0)); in TEST_P()
702 HloInstruction::CreateGetTupleElement(scalar_shape_, outer_param, 1)); in TEST_P()
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_unshard_params.py | 71 outer_param: Union[FlatParameter, nn.Parameter] = next(model.parameters())
73 param_to_check = outer_param if check_outer else inner_param
|
| /aosp_15_r20/external/rust/android-crates-io/crates/mockall_derive/src/ |
| D | lib.rs | 1021 'outer_param: for yparam in y.params.iter() { in merge_generics()
1039 continue 'outer_param; in merge_generics()
|