Home
last modified time | relevance | path

Searched defs:input_size (Results 1 – 25 of 470) sorted by relevance

12345678910>>...19

/aosp_15_r20/external/XNNPACK/test/
H A Dresize-bilinear-nhwc.cc103 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
118 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
134 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
150 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
260 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
276 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
293 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
310 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
424 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
440 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
[all …]
H A Dresize-bilinear-nchw.cc103 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
118 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
134 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
150 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
260 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
276 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
293 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
310 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
424 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
440 for (size_t input_size = 2; input_size <= 6; input_size += 2) { in TEST() local
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Drnn.cpp10 int64_t input_size, in RNNOptionsBase()
16 RNNOptions::RNNOptions(int64_t input_size, int64_t hidden_size) in RNNOptions()
19 LSTMOptions::LSTMOptions(int64_t input_size, int64_t hidden_size) in LSTMOptions()
22 GRUOptions::GRUOptions(int64_t input_size, int64_t hidden_size) in GRUOptions()
28 int64_t input_size, in RNNCellOptionsBase()
39 RNNCellOptions::RNNCellOptions(int64_t input_size, int64_t hidden_size) in RNNCellOptions()
42 LSTMCellOptions::LSTMCellOptions(int64_t input_size, int64_t hidden_size) in LSTMCellOptions()
45 GRUCellOptions::GRUCellOptions(int64_t input_size, int64_t hidden_size) in GRUCellOptions()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSample.h106 inline C10_UNUSED std::array<int64_t, 3> upsample_1d_common_check(IntArrayRef input_size, IntArrayR… in upsample_1d_common_check()
134 inline C10_UNUSED std::array<int64_t, 4> upsample_2d_common_check(IntArrayRef input_size, IntArrayR… in upsample_2d_common_check()
171 std::array<int64_t, 5> upsample_3d_common_check(IntArrayRef input_size, IntArrayRef output_size) { in upsample_3d_common_check()
256 int64_t input_size, in compute_scales_value()
267 int64_t input_size, in area_pixel_compute_scale()
314 int64_t input_size) { in nearest_neighbor_compute_source_index()
325 int64_t input_size) { in nearest_neighbor_exact_compute_source_index()
336 int64_t input_size, in nearest_idx()
357 int64_t input_size, in nearest_exact_idx()
437 inline void guard_index_and_lambda(const opmath_t& real_input_index, const int64_t& input_size, int… in guard_index_and_lambda()
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Drnn.h107 RNNImpl(int64_t input_size, int64_t hidden_size) in RNNImpl()
156 LSTMImpl(int64_t input_size, int64_t hidden_size) in LSTMImpl()
221 GRUImpl(int64_t input_size, int64_t hidden_size) in GRUImpl()
302 RNNCellImpl(int64_t input_size, int64_t hidden_size) in RNNCellImpl()
341 LSTMCellImpl(int64_t input_size, int64_t hidden_size) in LSTMCellImpl()
380 GRUCellImpl(int64_t input_size, int64_t hidden_size) in GRUCellImpl()
/aosp_15_r20/external/pytorch/benchmarks/fastrnns/
H A Dcustom_lstms.py36 input_size, argument
72 input_size, argument
123 def __init__(self, input_size, hidden_size): argument
183 def __init__(self, input_size, hidden_size, decompose_layernorm=False): argument
403 def test_script_rnn_layer(seq_len, batch, input_size, hidden_size): argument
423 def test_script_stacked_rnn(seq_len, batch, input_size, hidden_size, num_layers): argument
449 def test_script_stacked_bidir_rnn(seq_len, batch, input_size, hidden_size, num_layers): argument
481 seq_len, batch, input_size, hidden_size, num_layers argument
494 def test_script_stacked_lnlstm(seq_len, batch, input_size, hidden_size, num_layers): argument
/aosp_15_r20/trusty/user/base/app/swaes-benchmark/
Dvectors.h5583 size_t input_size; member
5599 #define INPUT_BYTES_DECRYPT(mode, key_size, input_size) \ argument
5602 #define INPUT_BYTES(mode, key_size, input_size, direction) \ argument
5605 #define OUTPUT_BYTES_ENCRYPT(mode, key_size, input_size) \ argument
5609 #define OUTPUT_BYTES(mode, key_size, input_size, direction) \ argument
5615 #define TAG(key_size, input_size) tag_##key_size##_##input_size argument
/aosp_15_r20/external/puffin/src/
H A Dbrotli_util.cc22 size_t input_size, in BrotliEncode()
59 size_t input_size, in BrotliEncode()
66 size_t input_size, in BrotliEncode()
73 size_t input_size, in BrotliDecode()
105 size_t input_size, in BrotliDecode()
/aosp_15_r20/trusty/user/base/app/hwaes-benchmark/
Dvectors.h5571 #define INPUT_BYTES_DECRYPT(mode, key_size, input_size) \ argument
5574 #define INPUT_BYTES(mode, key_size, input_size, direction) \ argument
5577 #define OUTPUT_BYTES_ENCRYPT(mode, key_size, input_size) \ argument
5581 #define OUTPUT_BYTES(mode, key_size, input_size, direction) \ argument
5587 #define TAG(key_size, input_size) tag_##key_size##_##input_size argument
/aosp_15_r20/external/openthread/third_party/mbedtls/repo/programs/psa/
H A Dcrypto_examples.c51 size_t input_size, in cipher_operation()
89 size_t input_size, in cipher_encrypt()
120 size_t input_size, in cipher_decrypt()
199 input_size = 100, in cipher_example_encrypt_decrypt_aes_cbc_pkcs7_multi() enumerator
247 input_size = 100, in cipher_example_encrypt_decrypt_aes_ctr_multi() enumerator
/aosp_15_r20/external/mbedtls/programs/psa/
H A Dcrypto_examples.c51 size_t input_size, in cipher_operation()
89 size_t input_size, in cipher_encrypt()
120 size_t input_size, in cipher_decrypt()
199 input_size = 100, in cipher_example_encrypt_decrypt_aes_cbc_pkcs7_multi() enumerator
247 input_size = 100, in cipher_example_encrypt_decrypt_aes_ctr_multi() enumerator
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DLinear.cpp33 IntArrayRef input_size, const Tensor& grad_output, const Tensor& weight) { in mkldnn_linear_backward_input()
98 auto input_size = self.sizes(); in mkldnn_linear() local
112 IntArrayRef input_size, const Tensor& grad_output, const Tensor& weight_t){ in mkldnn_linear_backward_input()
191 auto input_size = input.sizes(); in mkldnn_linear_pointwise() local
278 auto input_size = input.sizes(); in mkldnn_linear_pointwise_binary() local
370 auto input_size = self.sizes(); in mkl_linear() local
H A DMKLDNNConversions.cpp147 c10::OptionalArrayRef<int64_t> input_size) { in mkldnn_reorder_conv2d_weight()
204 c10::OptionalArrayRef<int64_t> input_size) { in mkldnn_reorder_conv3d_weight()
250 c10::OptionalArrayRef<int64_t> input_size) { in mkldnn_reorder_conv_weight()
267 ideep::dims input_size; in mkldnn_reorder_linear_weight() local
329 c10::OptionalArrayRef<int64_t> input_size) { in mkldnn_reorder_conv_transpose_weight()
469 c10::OptionalArrayRef<int64_t> input_size) { in mkldnn_reorder_mkldnn_rnn_layer_weight()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dunique_op_gpu.cu.h60 int64_t input_size, int64_t uniq_size, in ExtractFirstOccurrenceIndicesKernel()
86 Status ExtractFirstOccurrenceIndices(const GPUDevice& d, int64_t input_size, in ExtractFirstOccurrenceIndices()
145 int64_t input_size, const TIndex* sorted_input_inds, in LookupAndScatterUniqueIdsKernel()
158 Status LookupAndScatterUniqueIds(const GPUDevice& d, int64_t input_size, in LookupAndScatterUniqueIds()
211 int64_t input_size = input.NumElements(); in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dkernel_shape_util.cc20 Status GetWindowedOutputSizeVerboseV2(int64_t input_size, int64_t filter_size, in GetWindowedOutputSizeVerboseV2()
67 Status GetWindowedOutputSizeVerbose(int64_t input_size, int64_t filter_size, in GetWindowedOutputSizeVerbose()
78 Status GetWindowedOutputSize(int64_t input_size, int64_t filter_size, in GetWindowedOutputSize()
92 Status GetWindowedOutputSizeV2(int64_t input_size, int64_t filter_size, in GetWindowedOutputSizeV2()
/aosp_15_r20/external/eigen/bench/tensors/
H A Dtensor_benchmarks.h143 Eigen::array<TensorIndex, 2> input_size; in rowChip() local
164 Eigen::array<TensorIndex, 2> input_size; in colChip() local
381 Eigen::array<TensorIndex, 2> input_size; in rowReduction() local
413 Eigen::array<TensorIndex, 2> input_size; in colReduction() local
447 Eigen::array<TensorIndex, 2> input_size; in fullReduction() local
/aosp_15_r20/external/pigweed/pw_varint/
H A Dvarint_test_c.c42 size_t input_size, in pw_varint_CallDecode32()
48 size_t input_size, in pw_varint_CallDecode64()
54 size_t input_size, in pw_varint_CallZigZagAndVarintDecode64()
/aosp_15_r20/external/pytorch/test/quantization/bc/
H A Dtest_backward_compatibility.py77 input_size=None, argument
129 input_size=None, argument
188 self, obs, input_size, subname=None, generate=False, check_numerics=True argument
214 def _test_package(self, fp32_module, input_size, generate=False): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dkernel_utils.cc26 int input_size, int num_units, int batch_size, in RnnBatchStep()
42 int input_size, int aux_input_size, int num_units, in RnnBatchStep()
122 float recurrent_weights_scale, const float* bias_ptr, int input_size, in RnnBatchStep()
148 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep()
/aosp_15_r20/external/cronet/third_party/brotli/enc/
H A Dencode.c212 static size_t HashTableSize(size_t max_table_size, size_t input_size) { in HashTableSize()
221 size_t input_size, size_t* table_size) { in GetHashTable()
855 const size_t input_size, in CopyInputToRingBuffer()
1242 int lgwin, size_t input_size, const uint8_t* input_buffer, in BrotliCompressBufferQuality10()
1477 size_t BrotliEncoderMaxCompressedSize(size_t input_size) { in BrotliEncoderMaxCompressedSize()
1491 const uint8_t* input, size_t input_size, uint8_t* output) { in MakeUncompressedStream()
1523 int quality, int lgwin, BrotliEncoderMode mode, size_t input_size, in BrotliEncoderCompress()
1524 const uint8_t input_buffer[BROTLI_ARRAY_PARAM(input_size)], in BrotliEncoderCompress()
2095 size_t input_size) { in BrotliEncoderEstimatePeakMemoryUsage()
2236 const uint8_t* input, size_t input_size, uint8_t* output) { in MakeUncompressedStreamForTest()
/aosp_15_r20/external/pytorch/torch/nn/
H A Dgrad.py9 input_size, argument
107 input_size, argument
205 input_size, argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsvdf_test.cc131 BaseSVDFOpModel(int batches, int units, int input_size, int memory_size, in BaseSVDFOpModel()
180 int input_size() { return input_size_; } in input_size() function in tflite::__anona04cb8d10111::BaseSVDFOpModel
206 HybridSVDFOpModel(int batches, int units, int input_size, int memory_size, in HybridSVDFOpModel()
470 IntegerSVDFOpModel(int batches, int units, int input_size, int memory_size, in IntegerSVDFOpModel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSample.cuh97 int input_size, in area_pixel_compute_scale()
136 int input_size) { in nearest_neighbor_compute_source_index()
150 int input_size) { in nearest_neighbor_exact_compute_source_index()
311 const int input_size, in _compute_weights_span()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Ddepthwise_conv_op_base.py146 def Config(input_size, argument
182 def Config(input_size, argument
230 def Config(input_size, argument
264 def Config(input_size, argument
/aosp_15_r20/external/webrtc/modules/audio_processing/agc2/rnn_vad/
H A Drnn_gru.cc54 void ComputeUpdateResetGate(int input_size, in ComputeUpdateResetGate()
90 void ComputeStateGate(int input_size, in ComputeStateGate()
125 const int input_size, in GatedRecurrentLayer()

12345678910>>...19