Home
last modified time | relevance | path

Searched defs:input_one (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/
H A Dadd_test.py40 def forward(self, input_one, input_two): argument
68 def forward(self, input_one, mat1, mat2): argument
94 def forward(self, input_one, vec1, vec2): argument
134 def forward(self, input_one, batch1, batch2): argument
H A Dmatrix_mult_test.py52 def forward(self, input_one, input_two): argument
102 def forward(self, input_one, input_two): argument
H A Dbatchnorm_test.py68 def forward(self, input_one, mean, var, weight, bias, training, cudnn): argument
119 def forward(self, input_one, mean, var, weight, bias, training, cudnn): argument
H A Dfill_test.py46 def forward(self, input_one): argument
H A Dlinear_test.py18 def forward(self, input_one): argument
H A Dlinear_unpack_fp16_test.py39 def forward(self, input_one): argument
H A Dlinear_prepack_fp16_test.py36 def forward(self, input_one): argument
H A Dchunk_test.py33 def forward(self, input_one, chunks: int): argument
H A Dgather_test.py42 def forward(self, input_one, dim: int, index): argument
H A Dindex_select_test.py52 def forward(self, input_one, dim, index): argument
H A Dmatmul_test.py46 def forward(self, input_one, input_two): argument
H A Dhardswish_test.py44 def forward(self, input_one): argument
H A Dhardsigmoid_test.py44 def forward(self, input_one): argument
H A Das_strided_test.py47 self, input_one, size: List[int], stride: List[int], storage_offset: int argument
H A Dbinary_test.py95 def forward(self, input_one, input_two): argument
/aosp_15_r20/art/compiler/optimizing/
H A Dreference_type_propagation.cc253 HInstruction* input_one = compare->InputAt(0); in BoundTypeForClassCheck() local
H A Dinstruction_simplifier.cc1773 HInstruction* input_one = condition->InputAt(0); in RecognizeAndSimplifyClassCheck() local