| /aosp_15_r20/external/libopus/dnn/ |
| H A D | adaconvtest.c | 21 int feature_dim, in adaconv_compare()
123 int feature_dim, in adacomb_compare()
239 int feature_dim, in adashape_compare()
|
| H A D | nndsp.c | 139 int feature_dim, in adaconv_process_frame()
251 int feature_dim, in adacomb_process_frame()
346 int feature_dim, in adashape_process_frame()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | conv_op_helpers.cc | 107 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in CheckConvAttrs() local
210 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp() local
301 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp() local
|
| H A D | bias_ops.cc | 54 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile() local
|
| H A D | depthtospace_op.cc | 69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile() local
|
| H A D | extract_image_patches_op.cc | 62 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in Compile() local
|
| H A D | spacetodepth_op.cc | 69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | approx_topk_test.py | 119 def test_mips(self, k, db_size, qy_size, feature_dim): argument
146 def test_l2ann(self, k, db_size, qy_size, feature_dim): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| H A D | approx_topk_test.py | 121 def test_mips(self, dtype, k, db_size, qy_size, feature_dim): argument
147 def test_l2ann(self, dtype, k, db_size, qy_size, feature_dim): argument
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/activation_sparsifier/ |
| H A D | activation_sparsifier.py | 97 feature_dim=None, argument
192 feature_dim=None, argument
|
| /aosp_15_r20/external/libopus/dnn/torch/fargan/ |
| H A D | fargan.py | 157 def __init__(self, feature_dim=20, cond_size=256, pembed_dims=12): argument
274 …def __init__(self, subframe_size=40, nb_subframes=4, feature_dim=20, cond_size=256, passthrough_si… argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Embedding.cu | 170 int feature_dim = start_feature + ii * C10_WARP_SIZE; in embedding_backward_kernel() local
184 int feature_dim = start_feature + ii * C10_WARP_SIZE; in embedding_backward_kernel() local
|
| H A D | Indexing.cu | 93 int64_t feature_dim = start_feature + ii * C10_WARP_SIZE; in indexing_backward_kernel() local
113 int64_t feature_dim = start_feature + ii * C10_WARP_SIZE; in indexing_backward_kernel() local
256 int64_t feature_dim = start_feature + ii * C10_WARP_SIZE; in indexing_backward_kernel_quantized() local
269 int64_t feature_dim = start_feature + ii * C10_WARP_SIZE; in indexing_backward_kernel_quantized() local
|
| /aosp_15_r20/external/libopus/dnn/torch/rdovae/rdovae/ |
| H A D | rdovae.py | 302 def __init__(self, feature_dim, output_dim, cond_size, cond_size2, state_size=24): argument
519 feature_dim, argument
|
| /aosp_15_r20/external/libopus/dnn/torch/osce/utils/layers/ |
| H A D | noise_shaper.py | 39 feature_dim, argument
|
| H A D | td_shaper.py | 12 feature_dim, argument
|
| H A D | limited_adaptive_comb1d.py | 42 feature_dim, argument
|
| H A D | limited_adaptive_conv1d.py | 46 feature_dim, argument
|
| /aosp_15_r20/external/libopus/dnn/torch/osce/models/ |
| H A D | silk_feature_net.py | 40 feature_dim=47, argument
|
| H A D | lpcnet_feature_net.py | 39 feature_dim=84, argument
|
| H A D | silk_feature_net_pl.py | 47 feature_dim=47, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_shape_utils.cc | 125 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | cudnn_simplify_padding.cc | 92 int64_t feature_dim = dnums.kernel_output_feature_dimension(); in NumTrailingZeroOutputFeatures() local
|
| /aosp_15_r20/external/libopus/dnn/torch/neural-pitch/ |
| H A D | neural_pitch_update.py | 68 feature_dim = features.shape[1] variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 408 Value broadcast_from, int64_t feature_dim, in Broadcast1DToFeatureDim()
991 int64_t feature_dim = GetTensorFeatureDimIndex(num_dims, format); in GetConvDimensionNumbersAttr() local
1028 auto feature_dim = GetFeatureDimension(data_format, value_type); in matchAndRewrite() local
2236 auto feature_dim = feature_dim_attr.getValue().getSExtValue(); in matchAndRewrite() local
2345 auto feature_dim = getFeatureDimensionAttr(rewriter, data_format, op.x()); in matchAndRewrite() local
4998 const int feature_dim = in matchAndRewrite() local
5161 const int feature_dim = in matchAndRewrite() local
|