Searched defs:output_r (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Conv.cpp | 389 Tensor& output_r, in _convolution_out()
548 Tensor output_r; in _convolution() local
|
| /aosp_15_r20/external/libvpx/test/ |
| H A D | idct8x8_test.cc | 62 double output_r[64]; in TEST() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 559 const Tensor& output_r, const Tensor& grad_output_r, const Tensor& grad_hy, in miopen_rnn_backward_input()
675 const Tensor& output_r, in miopen_rnn_backward_weight()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | driver.cc | 313 std::regex output_r("output shape is \\|([^\\|]+)\\|,"); in ParseBufferAssignment() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 1674 const Tensor& output_r, in _cudnn_rnn_backward_input()
1907 const Tensor& output_r, in _cudnn_rnn_backward_weight()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 350 std::array<int32_t, max_nb_dims> output_l, output_r; in GetTrtBroadcastShape() local
|