Searched refs:outer_tuple (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
| H A D | Module.cpp | 1677 THPObjectPtr outer_tuple(PyTuple_New(result_size)); in THCPModule_cuda_tunableop_get_results() local
1678 if (!outer_tuple) in THCPModule_cuda_tunableop_get_results()
1703 outer_tuple.get(), result_index++, inner_tuple.release()); in THCPModule_cuda_tunableop_get_results()
1706 return outer_tuple.release(); in THCPModule_cuda_tunableop_get_results()
1717 THPObjectPtr outer_tuple(PyTuple_New(validators.size())); in THCPModule_cuda_tunableop_get_validators() local
1718 if (!outer_tuple) in THCPModule_cuda_tunableop_get_validators()
1734 outer_tuple.get(), validator_index++, inner_tuple.release()); in THCPModule_cuda_tunableop_get_validators()
1736 return outer_tuple.release(); in THCPModule_cuda_tunableop_get_validators()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_alias_analysis_test.cc | 757 auto outer_tuple = outer_builder.AddInstruction( in TEST_F() local
760 tuple_shape, condition1, inner_body, outer_tuple)); in TEST_F()
|
| H A D | hlo_dataflow_analysis_test.cc | 703 auto outer_tuple = outer_builder.AddInstruction( in TEST_P() local
706 tuple_shape, condition, inner_body, outer_tuple)); in TEST_P()
|
| H A D | hlo_evaluator_test.cc | 3210 auto outer_tuple = in TEST_P() local
3214 HloInstruction::CreateGetTupleElement(tuple2->shape(), outer_tuple, 1)); in TEST_P()
|