| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | QLSTMLayerNormalizationFixture.h | 48 …void setup(TensorShape input_shape, TensorShape weight_shape, TensorShape bias_shape, DataType dat… in setup()
100 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weight_shape, const T… in compute_target()
116 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weight_shape, cons… in compute_reference()
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/_expanded_weights/ |
| H A D | conv_utils.py | 210 weight_shape, argument
261 weight_shape, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | embedding_lookup_test.cc | 42 std::initializer_list<int> weight_shape, in BaseEmbeddingLookupOpModel()
91 std::initializer_list<int> weight_shape, in HybridEmbeddingLookupOpModel()
|
| H A D | lsh_projection_test.cc | 35 std::initializer_list<int> weight_shape) { in LSHProjectionOpModel()
|
| H A D | quant_basic_lstm_test.cc | 47 std::vector<int> weight_shape{4 * outputSize, outputSize + inputSize}; in QuantizedLSTMOpModel() local
|
| H A D | unidirectional_sequence_lstm.cc | 694 const RuntimeShape& weight_shape = GetTensorShape(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias() local
|
| H A D | lstm.cc | 1134 const RuntimeShape& weight_shape = GetTensorShape(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | quantize_weights_test.cc | 40 const TensorShape& weight_shape, in BuildGraphDef()
|
| H A D | fold_old_batch_norms_test.cc | 373 auto weight_shape = in TestFoldFusedBatchNormsWithConcat() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | lstm_utils_test.cc | 49 SmallVector<int64_t, 2> weight_shape{3, 12}; in createLstmCompositeFunc() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSHProjectionTest.cpp | 46 std::initializer_list<uint32_t> weight_shape) in LSHProjectionOpModel()
|
| D | EmbeddingLookupTest.cpp | 59 std::initializer_list<uint32_t> weight_shape) { in EmbeddingLookupOpModel()
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/mutators/ |
| H A D | InPlaceOperationMutator.cpp | 106 const auto weight_shape = weight_tensor->desc().shape; in try_in_place_depthwiseconv() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/ |
| H A D | test_executor.cpp | 130 std::vector<int> weight_shape; in patch_conv2d() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | conv_2d_helpers.cc | 211 RuntimeShape weight_shape = {weights_height_size, weights_width_size, in SplitWeightsForDwConv() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.cpp | 65 const Shape& weight_shape, const int32_t* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | xnnpack_test.cpp | 65 constexpr std::array<int64_t, 2u> weight_shape{41, 37}; in TEST() local
|
| H A D | vulkan_quantized_api_test.cpp | 1773 const at::IntArrayRef weight_shape, in test_quantized_conv_transpose2d()
2482 const at::IntArrayRef weight_shape, in test_quantized_conv2d()
3474 const at::IntArrayRef weight_shape, in test_quantized_linear()
|
| H A D | vulkan_api_test.cpp | 1476 const at::IntArrayRef weight_shape, in test_conv2d_context()
1517 const at::IntArrayRef weight_shape, in test_backwards_compatible_conv2d_context()
1558 const at::IntArrayRef weight_shape, in test_transposed_conv2d_context()
2732 const at::IntArrayRef weight_shape, in test_packed_layer_norm()
2840 const at::IntArrayRef weight_shape, in test_layer_norm()
2892 const at::IntArrayRef weight_shape, in test_native_layer_norm()
7841 const at::IntArrayRef weight_shape, in test_linear()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/ |
| H A D | pattern_utils_test.cc | 35 auto weight_shape = ops::Placeholder::Shape({32, 64}); in GetMatMulBiasAddGeluGraph() local
|
| /aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| H A D | onedevice_train_palm_detection_filter_inference_beta.cc | 187 const int32_t* __restrict weight_shape, in MatMul()
339 const int32_t* __restrict weight_shape, in FullyConnected()
|
| H A D | onedevice_train_palm_detection_filter_inference.cc | 188 const int32_t* __restrict weight_shape, in MatMul()
340 const int32_t* __restrict weight_shape, in FullyConnected()
|
| H A D | onedevice_train_palm_detection_filter_inference_v2.cc | 187 const int32_t* __restrict weight_shape, in MatMul()
339 const int32_t* __restrict weight_shape, in FullyConnected()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | mkl_remapper_test.cc | 757 auto weight_shape = in VerifyFused() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_test.cc | 2101 std::initializer_list<int> weight_shape) { in LSHProjectionOpModel()
4830 std::initializer_list<int> weight_shape, in BaseEmbeddingLookupOpModel()
|