Searched defs:input_contiguous (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LossMultiLabelMargin.cpp | 62 const Tensor& input_contiguous, in multilabel_margin_loss_forward_out_frame()
146 auto input_contiguous = input.contiguous(); in multilabel_margin_loss_forward_out_cpu_template() local
160 const Tensor& input_contiguous, in multilabel_margin_loss_backward_out_frame()
256 auto input_contiguous = input.contiguous(); in multilabel_margin_loss_backward_out_cpu_template() local
|
| H A D | LossMultiMargin.cpp | 125 auto input_contiguous = input.contiguous(); in multi_margin_loss_out_cpu_template() local
233 auto input_contiguous = input.contiguous(); in multi_margin_loss_backward_out_cpu_template() local
|
| H A D | LossNLL2d.cpp | 174 auto input_contiguous = input.contiguous(); in nll_loss2d_forward_out_frame() local
|
| H A D | LossNLL.cpp | 223 auto input_contiguous = input.contiguous(); in nll_loss_out_frame() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | quantization.py | 1949 def _get_linear_node(match, input_dim_exceeds_two, input_contiguous): argument
1972 linear_node, input_index, dtype, input_dim_exceeds_two, input_contiguous argument
2008 def _is_valid_dequant_linear_pattern(dtype, input_dim_exceeds_two, input_contiguous): argument
2085 input_contiguous=True, argument
2335 input_contiguous=True, argument
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | TestUtil.h | 34 tf, op, input_contiguous, expected_contiguous, channels_last_support) \ argument
72 tf, op, input_contiguous, expected_contiguous, channels_last_support) \ argument
|
| H A D | op_abs_test.cpp | 47 Tensor input_contiguous = in TEST_F() local
|