| /external/ComputeLibrary/tests/validation/fixtures/ |
| 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()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_test.cc | 42 std::initializer_list<int> weight_shape, in BaseEmbeddingLookupOpModel()
91 std::initializer_list<int> weight_shape, in HybridEmbeddingLookupOpModel()
|
| D | lsh_projection_test.cc | 35 std::initializer_list<int> weight_shape) { in LSHProjectionOpModel()
|
| D | quant_basic_lstm_test.cc | 47 std::vector<int> weight_shape{4 * outputSize, outputSize + inputSize}; in QuantizedLSTMOpModel() local
|
| D | unidirectional_sequence_lstm.cc | 694 const RuntimeShape& weight_shape = GetTensorShape(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias() local
|
| D | lstm.cc | 1134 const RuntimeShape& weight_shape = GetTensorShape(weight_tensor); in PrecomputeZeroPointTimesWeightWithBias() local
|
| D | lstm_eval.cc | 55 tflite::RuntimeShape weight_shape({m_rows, m_cols}); in MatrixBatchVectorMultiplyAccumulate() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_weights_test.cc | 40 const TensorShape& weight_shape, in BuildGraphDef()
|
| D | fold_old_batch_norms_test.cc | 373 auto weight_shape = in TestFoldFusedBatchNormsWithConcat() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils_test.cc | 49 SmallVector<int64_t, 2> weight_shape{3, 12}; in createLstmCompositeFunc() local
|
| /external/ComputeLibrary/src/graph/mutators/ |
| D | InPlaceOperationMutator.cpp | 106 const auto weight_shape = weight_tensor->desc().shape; in try_in_place_depthwiseconv() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_helpers.cc | 211 RuntimeShape weight_shape = {weights_height_size, weights_width_size, in SplitWeightsForDwConv() local
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | pattern_utils_test.cc | 35 auto weight_shape = ops::Placeholder::Shape({32, 64}); in GetMatMulBiasAddGeluGraph() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | mkl_remapper_test.cc | 757 auto weight_shape = in VerifyFused() local
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| 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()
|
| 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()
|
| 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()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2101 std::initializer_list<int> weight_shape) { in LSHProjectionOpModel()
4830 std::initializer_list<int> weight_shape, in BaseEmbeddingLookupOpModel()
|