| /aosp_15_r20/external/libopus/dnn/torch/neural-pitch/ |
| H A D | models.py | 11 def __init__(self, input_dim=88, gru_dim=64, output_dim=192): argument
35 def __init__(self, input_dim=90, gru_dim=64, output_dim=192): argument
75 def __init__(self,input_IF_dim=88, input_xcorr_dim=224, gru_dim=64, output_dim=192): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | indexed_array_analysis.cc | 220 for (int64_t output_dim : output_dims) { in FoldGatherOfGather() local
543 for (int64_t output_dim : operand->output_dims()) { in ReshapeToAddDegenerateDims() local
570 for (int64_t output_dim : new_output_dims) { in ReshapeToAddDegenerateDims() local
717 int64_t output_dim = scalar_indexed->output_dims()[i]; in FoldReshapeOfGatherNoDegenerateDims() local
881 auto is_broadcasted_dim = [&](int64_t output_dim) { in ComputeArrayForElementwiseBinaryOp()
|
| H A D | dynamic_padder.cc | 367 const int64_t output_dim = output_dims[i]; in GenerateBinaryMask() local
573 for (int64_t output_dim : output_dims) { in RewriteDynamicReshapeSplitInput() local
653 int64_t output_dim, absl::Span<HloInstruction*> input_dynamic_dims, in RewriteDynamicReshapeCombineInput()
795 int64_t output_dim = output_dims[0]; in RewriteDynamicReshapeSingleGroup() local
|
| /aosp_15_r20/external/libopus/dnn/torch/rdovae/rdovae/ |
| H A D | rdovae.py | 263 def __init__(self, input_dim, output_dim, dilation=1): argument
302 def __init__(self, feature_dim, output_dim, cond_size, cond_size2, state_size=24): argument
382 def __init__(self, input_dim, output_dim, cond_size, cond_size2, state_size=24): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | image_ops.cc | 227 DimensionHandle output_dim; in CombinedNMSShapeFn() local
1026 DimensionHandle output_dim; in __anon269972ec1b02() local
1053 DimensionHandle output_dim; in __anon269972ec1c02() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorTransformerFunctions.cu | 45 int output_dim, in remove_padding_transform0213_2()
86 int output_dim, in remove_padding_2()
119 int output_dim, in remove_padding()
157 int output_dim, in remove_padding_kernelLauncher()
191 int output_dim, in remove_padding_transform0213_kernelLauncher()
|
| /aosp_15_r20/external/libopus/dnn/torch/lpcnet/utils/layers/ |
| H A D | dual_fc.py | 34 def __init__(self, input_dim, output_dim): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | batch_to_space_nd_test.cc | 25 int input_dim, int output_dim) { in GetIndexRange()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | reshape_tester.h | 46 for (int32_t output_dim : output_shape) { in OutputShape() local
|
| H A D | pad_tester.cc | 42 int32_t output_dim = InputShape()[i]; in OutputShape() local
|
| H A D | quantized_pad_tester.cc | 41 int32_t output_dim = InputShape()[i]; in OutputShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/debug/examples/v1/ |
| H A D | debug_mnist_v1.py | 158 def nn_layer(input_tensor, input_dim, output_dim, layer_name, act=tf.nn.relu): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/debug/examples/v2/ |
| H A D | debug_mnist_v2.py | 160 def get_dense_weights(input_dim, output_dim): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/ |
| H A D | embeddings.py | 108 output_dim, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | argmax_spmd_expander.cc | 158 int output_dim = 0; in ComputeLayoutBackward() local
|
| H A D | reduce_spmd_expander.cc | 301 int output_dim = 0; in ComputeLayoutBackward() local
|
| /aosp_15_r20/external/pytorch/benchmarks/gpt_fast/ |
| H A D | benchmark.py | 37 def __init__(self, input_dim, hidden_dim, output_dim, dtype): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | matrix.cc | 500 if (auto output_dim = output_dimension_number(x_config[d])) { in Einsum() local
508 if (auto output_dim = output_dimension_number(y_config[d])) { in Einsum() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | spacetobatch_op.cc | 81 const int64_t output_dim = in SpaceToBatch() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/gradient/ |
| H A D | bcast_grad_args.cc | 140 int output_dim = -1; in Invoke() local
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_spawn_gloo.py | 189 def __init__(self, input_dim, hidden_dim, output_dim, hidden_layers): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_symbolic_shape_optimization.cc | 239 int64_t output_dim = out_type.getShape()[d]; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| H A D | replace_cast_hacks_with_tf_xla_ops.cc | 77 int output_dim, const SmallVector<int64_t> &weight_non_output_indices) { in CalculateZeroPointOffset()
|
| /aosp_15_r20/external/executorch/kernels/quantized/cpu/ |
| H A D | op_choose_qparams.cpp | 263 size_t output_dim = input.dim(); in choose_qparams_per_token_asymmetric_out() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | bcast.h | 202 int output_dim = -1; in BCastList() local
|