| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Embedding.cpp | 57 const Tensor & grad, const Tensor & indices, c10::SymInt num_weights, in embedding_backward_symint()
76 const Tensor & grad_, const Tensor & indices_, int64_t num_weights, in embedding_sparse_backward()
113 const Tensor & grad_, const Tensor & indices, int64_t num_weights, in embedding_dense_backward_cpu()
|
| H A D | EmbeddingBag.cpp | 1379 int64_t num_weights, in _embedding_bag_backward()
1394 c10::SymInt num_weights, in _embedding_bag_backward_symint()
1449 int64_t num_weights) { in _embedding_bag_dense_backward_cpu_max()
1465 int64_t num_weights, in compute_counts()
1486 int64_t num_weights, in compute_counts_uniq()
1509 int64_t num_weights, in _embedding_bag_dense_backward_cpu_sum_mean()
1597 const Tensor& max_indices_, int64_t num_weights, in _embedding_bag_dense_backward_cpu()
1752 const Tensor &offset2bag, const Tensor &bag_size_, SymInt num_weights, in _embedding_bag_sparse_backward_symint()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | EmbeddingBag.cu | 173 int64_t num_weights, in embedding_bag_backward_cuda_sum_avg()
270 int64_t num_weights, in embedding_bag_backward_cuda_max()
431 int64_t num_weights, in _embedding_bag_dense_backward_cuda()
|
| H A D | Embedding.cu | 249 int64_t num_weights, int64_t padding_idx, in embedding_dense_backward_cuda()
|
| H A D | EmbeddingBackwardKernel.cu | 217 int64_t num_weights, in embedding_backward_cuda_kernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/microfrontend/lib/ |
| H A D | filterbank_io.c | 39 int num_weights = 0; in FilterbankWriteMemmapPreamble() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | count_ops.cc | 29 auto num_weights = c->NumElements(weights); in DenseCountSparseOutputShapeFn() local
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
| H A D | FullyConnectedLayer.cpp | 60 unsigned int num_weights = 1; in compute_weights_descriptor() local
|
| /aosp_15_r20/external/libpng/ |
| H A D | pngwrite.c | 1145 int num_weights, png_const_doublep filter_weights, in png_set_filter_heuristics()
1159 int num_weights, png_const_fixed_point_p filter_weights, in png_set_filter_heuristics_fixed()
|
| /aosp_15_r20/external/webrtc/modules/audio_processing/agc2/rnn_vad/ |
| H A D | spectral_features_internal_unittest.cc | 37 const int num_weights = std::accumulate(kOpusScaleNumBins24kHz20ms.begin(), in ComputeTriangularFiltersWeights() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sdca_ops.cc | 277 int64_t num_weights = 0; in Compute() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 57 c10::SymInt num_weights, c10::SymInt padding_idx, bool scale_grad_by_freq) { in embedding_dense_backward_batch_rule()
|
| /aosp_15_r20/external/swiftshader/third_party/astc-encoder/Source/ |
| H A D | astc_codec_internals.h | 80 int num_weights; member
|
| /aosp_15_r20/external/ethtool/ |
| H A D | ethtool.c | 4040 char **rxfhindir_weight, u32 num_weights) in fill_indir_table()
4089 char **rxfhindir_weight, u32 num_weights) in do_srxfhindir()
4149 u32 num_weights = 0; in do_srxfh() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | rnn.cpp | 223 int64_t num_weights = options_base.bias() ? 4 : 2; in flatten_parameters() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 1290 auto num_weights = in copy_weights_to_flat_buf_views() local
1486 auto num_weights = in _cudnn_rnn() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 499 auto num_weights = get_num_weights(handle, descs.rnn_desc, descs.x_descs[0], datatype); in miopen_rnn() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.cpp | 419 int64_t num_weights, in compute_shape_embedding_dense_backward()
|
| /aosp_15_r20/external/mesa3d/src/mesa/main/ |
| H A D | texcompress_astc.cpp | 600 int num_weights; member
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 6356 num_weights, argument
6398 num_weights, argument
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 11262 int num_weights = 32; in TEST_F() local
|