| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | SoftMax.cpp | 346 auto input_ = input.contiguous(); in TORCH_IMPL_FUNC() local
376 auto input_ = input.contiguous(); in TORCH_IMPL_FUNC() local
443 Tensor softmax(const Tensor& input_, const int64_t dim_, std::optional<ScalarType> dtype) { in softmax()
458 const Tensor& input_, in softmax_out()
495 Tensor special_softmax(const Tensor& input_, const int64_t dim_, std::optional<ScalarType> dtype) { in special_softmax()
499 Tensor log_softmax(const Tensor& input_, const int64_t dim_, std::optional<ScalarType> dtype) { in log_softmax()
514 const Tensor& input_, in log_softmax_out()
572 Tensor masked_softmax_cpu(const Tensor& input_, const Tensor& mask_, const std::optional<int64_t> d… in masked_softmax_cpu()
|
| H A D | Convolution.cpp | 916 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv1d_symint()
941 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv2d_symint()
966 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv3d_symint()
1075 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias, in conv1d_padding_symint()
1089 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias, in conv2d_padding_symint()
1103 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias, in conv3d_padding_symint()
1117 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv_transpose1d_symint()
1136 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv_transpose2d_symint()
1155 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv_transpose3d_symint()
1993 const Tensor& grad_output_, const Tensor& input_, const Tensor& weight_, in convolution_backward()
|
| H A D | NaiveDilatedConvolution.cpp | 555 const Tensor input_ = in slow_conv_dilated2d_cpu() local
607 const Tensor input_ = in slow_conv_dilated3d_cpu() local
659 const Tensor input_ = in slow_conv_dilated2d_backward_cpu() local
715 const Tensor input_ = in slow_conv_dilated3d_backward_cpu() local
|
| H A D | NaiveConvolutionTranspose2d.cpp | 215 Tensor input_ = input.contiguous(memory_format); in TORCH_META_FUNC() local
271 Tensor input_ = input.contiguous(memory_format); in slow_conv_transpose2d_out_cpu_template() local
390 const Tensor& input_, in slow_conv_transpose2d_backward_out_cpu_template()
587 const Tensor& input_, in slow_conv_transpose2d_acc_grad_parameters_cpu()
|
| /aosp_15_r20/external/python/asn1crypto/asn1crypto/ |
| D | _teletex_codec.py | 16 def encode(self, input_, errors='strict'): argument
19 def decode(self, input_, errors='strict'): argument
25 def encode(self, input_, final=False): argument
31 def decode(self, input_, final=False): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_test.cc | 590 int input_; member in tflite::__anon1a74836d0111::FloatPoolingOpModel
701 int input_; member in tflite::__anon1a74836d0111::ConvolutionOpModel
1079 int input_; member in tflite::__anon1a74836d0111::PerChannelQuantizedConvolutionWithConstantFilterOpModel
1194 int input_; member in tflite::__anon1a74836d0111::DepthwiseConvolutionOpModel
1319 int input_; member in tflite::__anon1a74836d0111::FullyConnectedOpModel
1419 int input_; member in tflite::__anon1a74836d0111::SoftmaxOpModel
1515 int input_; member in tflite::__anon1a74836d0111::ReshapeOpModel
1548 int input_; member in tflite::__anon1a74836d0111::SqueezeOpModel
1604 int input_; member in tflite::__anon1a74836d0111::L2NormOpModel
1642 int input_; member in tflite::__anon1a74836d0111::TransposeSimpleModel
[all …]
|
| H A D | nnapi_delegate_signed_quantization_test.cc | 122 int input_; member in tflite::__anonf5e0ceac0111::HybridFullyConnectedOpModel
311 int input_; member in tflite::__anonf5e0ceac0111::BaseConvolutionOpModel
724 int input_; member in tflite::__anonf5e0ceac0111::QuantizeOpModel
824 int input_; member in tflite::__anonf5e0ceac0111::DequantizeOpModel
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SoftMax.cpp | 536 const Tensor& input_, in softmax_sparse_cpu()
553 const Tensor& input_, in log_softmax_sparse_cpu()
573 const Tensor& input_) { in softmax_backward_sparse_cpu()
593 const Tensor& input_) { in log_softmax_backward_sparse_cpu()
609 Tensor _sparse_softmax(const Tensor& input_, const int64_t dim_, std::optional<ScalarType> dtype) { in _sparse_softmax()
627 Tensor _sparse_log_softmax(const Tensor& input_, const int64_t dim_, std::optional<ScalarType> dtyp… in _sparse_log_softmax()
|
| H A D | ParamUtils.cpp | 17 const Tensor& input_, in softmax_sparse_input_preprocessing()
37 const Tensor& input_, in softmax_backward_sparse_input_preprocessing()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| H A D | ops_defs.py | 53 def _composite_conv_add_relu(input_, filter_, bias, stride_w, stride_h, argument
122 def _composite_fully_connected(input_, filter_, bias, act): argument
172 def _composite_max_pool(input_, stride_w, stride_h, filter_width, filter_height, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | complex_support_test.cc | 48 int input_; member in tflite::__anon9b23ced20111::RealOpModel
109 int input_; member in tflite::__anon9b23ced20111::ImagOpModel
172 int input_; member in tflite::__anon9b23ced20111::ComplexAbsOpModel
|
| H A D | random_uniform_test.cc | 72 int input_; member in tflite::__anonee2af3de0111::RandomUniformOpModel
98 int input_; member in tflite::__anonee2af3de0111::RandomUniformIntOpModel
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | NaiveDilatedConvolution.cu | 427 const Tensor input_ = in slow_conv_dilated2d_cuda() local
476 const Tensor input_ = in slow_conv_dilated2d_backward_cuda() local
533 const Tensor input_ = in slow_conv_dilated3d_cuda() local
582 const Tensor input_ = in slow_conv_dilated3d_backward_cuda() local
|
| /aosp_15_r20/external/libaom/test/ |
| H A D | selfguided_filter_test.cc | 56 uint8_t *input_ = in RunSpeedTest() local
142 uint8_t *input_ = in RunCorrectnessTest() local
253 uint16_t *input_ = in RunSpeedTest() local
341 uint16_t *input_ = in RunCorrectnessTest() local
|
| H A D | warp_filter_test_util.cc | 149 std::unique_ptr<uint8_t[]> input_(new (std::nothrow) uint8_t[h * stride]); in RunSpeedTest() local
212 std::unique_ptr<uint8_t[]> input_(new (std::nothrow) uint8_t[h * stride]); in RunCheckOutput() local
345 std::unique_ptr<uint16_t[]> input_(new (std::nothrow) uint16_t[h * stride]); in RunSpeedTest() local
410 std::unique_ptr<uint16_t[]> input_(new (std::nothrow) uint16_t[h * stride]); in RunCheckOutput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | rnn_test.py | 58 def call(self, input_, state, scope=None): argument
76 def call(self, input_, state, scope=None): argument
94 def call(self, input_, state, scope=None): argument
115 def call(self, input_, state, scope=None): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesConvolution.cpp | 260 const auto input_ = reshape_dim_into(*input_bdim, 1, input); in convolution_backward_weight_batch_rule() local
314 const auto input_ = reshape_dim_into(*input_bdim, 1, input); in convolution_backward_weight_batch_rule() local
324 auto input_ = moveBatchDimToFront(input, input_bdim); // BN(GI) in convolution_backward_weight_batch_rule() local
362 const Tensor& grad_output_, const Tensor& input_, const Tensor& weight_, in convolution_backward_plumbing()
|
| H A D | BatchRulesNorm.cpp | 79 auto input_ = moveBatchDimToFront(input, input_bdim); in batch_norm_batch_rule() local
151 auto input_ = moveBatchDimToFront(input, input_bdim); in batch_norm_backward_no_weight_bias_batch_rule() local
311 const auto input_ = reshape_dim_into(*input_bdim, 0, input_value); in native_group_norm_plumbing() local
347 auto input_ = moveBatchDimToFront(input, input_bdim); in group_norm_backward_no_weight_bias_batch_rule() local
494 auto input_ = moveBatchDimToFront(input, input_bdim); in native_layer_norm_batch_rule() local
548 auto input_ = moveBatchDimToFront(input, input_bdim); in native_layer_norm_backward_no_weight_bias_batch_rule() local
|
| /aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/nom-7.1.3/src/multi/ |
| H A D | mod.rs | 463 let input_ = input.clone(); in many0_count() localVariable
527 let input_ = input.clone(); in many1_count() localVariable
582 let input_ = input.clone(); in count() localVariable
635 let input_ = input.clone(); in fill() localVariable
1032 let input_ = input.clone(); in length_count() localVariable
|
| /aosp_15_r20/external/rust/android-crates-io/crates/nom/src/multi/ |
| D | mod.rs | 463 let input_ = input.clone(); in many0_count() localVariable
527 let input_ = input.clone(); in many1_count() localVariable
582 let input_ = input.clone(); in count() localVariable
635 let input_ = input.clone(); in fill() localVariable
1032 let input_ = input.clone(); in length_count() localVariable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| H A D | ops_defs.py | 48 def _composite_mirror_pad(input_, paddings, mode): argument
98 def _composite_mirror_pad_grad(input_, paddings, mode): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | AdaptiveAvgPoolKernel.cpp | 20 const Tensor& input_, in cpu_adaptive_avg_pool2d()
74 const Tensor& input_, in cpu_adaptive_avg_pool2d_channels_last()
161 const Tensor& input_, in cpu_adaptive_avg_pool2d_channels_last()
415 const Tensor& input_, in cpu_adaptive_avg_pool3d()
480 const Tensor& input_, in cpu_adaptive_avg_pool3d_channels_last()
576 const Tensor& input_, in cpu_adaptive_avg_pool3d_channels_last()
|
| H A D | AdaptiveMaxPoolKernel.cpp | 21 const Tensor& input_, in cpu_adaptive_max_pool2d()
89 const Tensor& input_, in cpu_adaptive_max_pool2d_channels_last()
206 const Tensor& input_, in cpu_adaptive_max_pool2d_channels_last()
487 const Tensor& input_, in cpu_adaptive_max_pool3d()
563 const Tensor& input_, in cpu_adaptive_max_pool3d_channels_last()
688 const Tensor& input_, in cpu_adaptive_max_pool3d_channels_last()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SoftMax.cu | 588 const Tensor& input_, in softmax_sparse_cuda()
605 const Tensor& input_, in log_softmax_sparse_cuda()
625 const Tensor& input_) { in softmax_backward_sparse_cuda()
645 const Tensor& input_) { in log_softmax_backward_sparse_cuda()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | embedding.h | 32 auto input_ = input; in embedding() local
101 auto input_ = input; in embedding_bag() local
|