Searched defs:input_r (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Conv.cpp | 314 const Tensor& input_r, in get_onednn_conv_sum_attr()
390 const Tensor& input_r, in _convolution_out()
538 const Tensor& input_r, in _convolution()
564 const Tensor& input_r, in convolution_overrideable()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 31 const Tensor& input_r, TensorList weight, int64_t weight_stride0, in miopen_rnn()
446 const Tensor& input_r, TensorList weight, int64_t weight_stride0, in miopen_rnn()
558 const Tensor& input_r, const Tensor& weight_buf, const Tensor& hx, const Tensor& cx, in miopen_rnn_backward_input()
673 const Tensor& input_r, TensorList weight_arr, int64_t weight_stride0, in miopen_rnn_backward_weight()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/miopen/ |
| H A D | AutocastRNN.cpp | 13 miopen_rnn(const Tensor & input_r, in miopen_rnn()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 51 const Tensor& input_r, in _cudnn_rnn()
1382 const Tensor& input_r, in _cudnn_rnn()
1670 const Tensor& input_r, in _cudnn_rnn_backward_input()
1901 const Tensor& input_r, in _cudnn_rnn_backward_weight()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Convolution.cpp | 1312 const Tensor& input_r, const Tensor& weight_r, const std::optional<Tensor>& bias_opt, in select_conv_backend()
1469 const Tensor& input_r, const Tensor& weight_r, const std::optional<Tensor>& bias_r_opt, in _convolution()
1709 const Tensor& input_r, const Tensor& weight_r, const std::optional<Tensor>& bias_r_opt, in _convolution()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesConvolution.cpp | 127 const Tensor& input_r, const Tensor& weight_r, const std::optional<Tensor>& bias_r_opt, in _convolution_decomp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 4716 const TRT_TensorOrWeights& input_r) { in ConvertBatchMatMul()
4745 auto input_r = std::make_unique<TRT_TensorOrWeights>(inputs.at(1)); in ConvertBatchMatMul() local
|
| H A D | convert_nodes_test.cc | 3691 auto input_r = ops::Placeholder(s.WithOpName("input2"), dtype); in GetBinaryOpNodeDef() local
|