Home
last modified time | relevance | path

Searched refs:input_size_ (Results 1 – 25 of 26) sorted by relevance

12

/external/webrtc/modules/audio_processing/agc2/rnn_vad/
Drnn.h52 size_t input_size() const { return input_size_; } in input_size()
60 const size_t input_size_;
84 size_t input_size() const { return input_size_; } in input_size()
93 const size_t input_size_;
Drnn.cc275 : input_size_(input_size), in FullyConnectedLayer()
286 RTC_DCHECK_EQ(input_size_ * output_size_, weights_.size()) in FullyConnectedLayer()
300 ComputeFullyConnectedLayerOutputSse2(input_size_, output_size_, input, in ComputeOutput()
308 ComputeFullyConnectedLayerOutput(input_size_, output_size_, input, bias_, in ComputeOutput()
313 ComputeFullyConnectedLayerOutput(input_size_, output_size_, input, bias_, in ComputeOutput()
325 : input_size_(input_size), in GatedRecurrentLayer()
337 RTC_DCHECK_EQ(kNumGruGates * input_size_ * output_size_, weights_.size()) in GatedRecurrentLayer()
361 ComputeGruLayerOutput(input_size_, output_size_, input, weights_, in ComputeOutput()
368 ComputeGruLayerOutput(input_size_, output_size_, input, weights_, in ComputeOutput()
373 ComputeGruLayerOutput(input_size_, output_size_, input, weights_, in ComputeOutput()
/external/tensorflow/tensorflow/lite/kernels/
Dsvdf_test.cc138 input_size_(input_size), in BaseSVDFOpModel()
154 {batches_, input_size_}, // input tensor in BaseSVDFOpModel()
155 {units_ * rank, input_size_}, // weights_feature tensor in BaseSVDFOpModel()
180 int input_size() { return input_size_; } in input_size()
194 int input_size_; member in tflite::__anon7126f9820111::BaseSVDFOpModel
474 input_size_(input_size), in IntegerSVDFOpModel()
531 int input_size_; member in tflite::__anon7126f9820111::IntegerSVDFOpModel
Dlog_softmax_test.cc35 : batches_(batches), input_size_(size) { in LogSoftmaxOpModel()
40 BuildInterpreter({{batches_, input_size_}}); in LogSoftmaxOpModel()
58 int input_size_; member in tflite::__anoneb2c93520111::LogSoftmaxOpModel
Dunidirectional_sequence_rnn_test.cc181 input_size_(size) { in UnidirectionalRNNOpModel()
195 BuildInterpreter({{sequence_len_, batches_, input_size_}, in UnidirectionalRNNOpModel()
196 {units_, input_size_}, in UnidirectionalRNNOpModel()
201 BuildInterpreter({{batches_, sequence_len_, input_size_}, in UnidirectionalRNNOpModel()
202 {units_, input_size_}, in UnidirectionalRNNOpModel()
229 int input_size() { return input_size_; } in input_size()
245 int input_size_; member in tflite::__anone81b2e4e0111::UnidirectionalRNNOpModel
Dbasic_rnn_test.cc177 : batches_(batches), units_(units), input_size_(size) { in RNNOpModel()
188 BuildInterpreter({{batches_, input_size_}, // input tensor in RNNOpModel()
189 {units_, input_size_}, // weights tensor in RNNOpModel()
215 int input_size() { return input_size_; } in input_size()
229 int input_size_; member in tflite::__anon3123f5fe0111::RNNOpModel
Dsoftmax_test.cc36 : batches_(batches), input_size_(size), beta_(beta) { in SoftmaxOpModel()
41 BuildInterpreter({{batches_, input_size_}}); in SoftmaxOpModel()
59 int input_size_; member in tflite::__anon8d5b88b10111::SoftmaxOpModel
Dfully_connected_test.cc150 input_size_ = total_input_size / batches_;
154 weights_ = AddInput({TensorType_INT8, {units_, input_size_}, -63.5, 64});
157 AddInput({input.type, {units_, input_size_}, input.min, input.max});
199 int input_size() { return input_size_; } in input_size()
211 int input_size_; member in tflite::__anonae46fc4d0111::BaseFullyConnectedOpModel
306 input_size_ = total_input_size / batches_;
343 int input_size() { return input_size_; } in input_size()
355 int input_size_; member in tflite::__anonae46fc4d0111::HybridFullyConnectedOpModel
1196 input_size_ = total_input_size / batches_; in SparseFullyConnectedOpModel()
1222 int input_size() { return input_size_; } in input_size()
[all …]
Dquant_basic_lstm_test.cc41 input_size_ = inputSize; in QuantizedLSTMOpModel()
129 int inputSize() { return input_size_; } in inputSize()
156 int input_size_; member in tflite::__anond96182980111::QuantizedLSTMOpModel
Dbatch_matmul_test.cc361 input_size_ = total_input_size / batches_;
387 int input_size() { return input_size_; } in input_size()
400 int input_size_; member in tflite::__anon83a19b600111::HybridBatchMatMulOpModel
720 input_size_ = total_input_size / batches_;
724 {input_size_, units_},
765 int input_size_; member in tflite::__anon83a19b600111::QuantizedBatchMatMulOpModel
Dbidirectional_sequence_rnn_test.cc674 input_size_(input_size), in BidirectionalRNNOpModel()
690 (time_major) ? std::vector<int>({sequence_len_, batches_, input_size_}) in BidirectionalRNNOpModel()
691 : std::vector<int>({batches_, sequence_len_, input_size_}); in BidirectionalRNNOpModel()
731 {fw_units_, input_size_}, // fw_weights in BidirectionalRNNOpModel()
735 {bw_units_, input_size_}, // bw_weights in BidirectionalRNNOpModel()
816 int input_size() { return input_size_; } in input_size()
843 int input_size_; member in tflite::__anon7adf078d0111::BidirectionalRNNOpModel
Dlstm_eval_test.cc151 std::vector<int32_t> input_size_ = {n_batch_, n_input_}; member in tflite::__anon39d466b60111::BaseLstmParam
324 PackWeightToTensor(&input_tensor_, input_, input_size_); in GetInput()
683 PackWeightToTensor(&input_quantized_tensor_, input_quantized_, input_size_); in GetInputQuantized()
722 PackWeightToTensor(&input_tensor_, input_float_, input_size_); in GetFloatInput()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_tester.h39 input_size_ = ReshapeTester::ComputeSize(input_shape); in InputShape()
59 inline int32_t InputSize() const { return input_size_; } in InputSize()
79 int32_t input_size_ = 1; variable
Dfully_connected_tester.h41 input_size_ = ComputeSize(input_shape_); in InputShape()
47 inline int32_t InputSize() const { return input_size_; } in InputSize()
108 int32_t input_size_ = 1; variable
Dreduce_tester.h41 input_size_ = ReduceTester::ComputeSize(input_shape_); in InputShape()
47 inline int32_t InputSize() const { return input_size_; } in InputSize()
109 int32_t input_size_; variable
/external/tensorflow/tensorflow/core/kernels/rnn/
Dgru_ops.h33 input_size_(input_size), in GRUCell()
39 return {batch_size_, input_size_}; in x_extends()
43 return {0, input_size_}; in h_offsets()
64 const int input_size_; member
Dlstm_ops.h107 input_size_(input_size), in LSTMBlockCell()
112 int input_size() const { return input_size_; } in input_size()
143 return {batch_size_, input_size_}; in xh_x_extents()
147 return {0, input_size_}; in xh_h_offsets()
156 const int input_size_; member
Dlstm_ops_gpu.cu.cc428 cell_size_, input_size_); \
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dmatmul_test.cc35 input_size_ = total_input_size / batches_; in FullyConnectedOpModel()
39 AddInput({input.type, {units_, input_size_}, input.min, input.max}); in FullyConnectedOpModel()
104 int input_size_; member in tflite::FullyConnectedOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_signed_quantization_test.cc75 input_size_ = total_input_size / batches_;
113 int input_size() { return input_size_; } in input_size()
125 int input_size_; member in tflite::__anonc4d0941d0111::HybridFullyConnectedOpModel
Dnnapi_delegate_test.cc2738 : batches_(batches), units_(units), input_size_(size) { in RNNOpModel()
2749 {batches_, input_size_}, // input tensor in RNNOpModel()
2750 {units_, input_size_}, // weights tensor in RNNOpModel()
2777 int input_size() { return input_size_; } in input_size()
2791 int input_size_; member in tflite::__anonde5f765e0111::RNNOpModel
2925 input_size_(input_size), in BaseSVDFOpModel()
2942 {batches_, input_size_}, // input tensor in BaseSVDFOpModel()
2943 {units_ * rank, input_size_}, // weights_feature tensor in BaseSVDFOpModel()
2970 int input_size() { return input_size_; } in input_size()
2984 int input_size_; member in tflite::__anonde5f765e0111::BaseSVDFOpModel
/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py2420 for (name_, input_size_, filter_size_, output_size_, stride_,
2423 GetMaxPoolFwdTest(input_size_, filter_size_, stride_, padding_))
2425 GetMaxPoolGradTest(input_size_, filter_size_, output_size_, stride_,
2428 GetMaxPoolGradGradTest(input_size_, filter_size_, output_size_,
Dconv_ops_test.py3438 for index, (input_size_, filter_size_, output_size_, stride_,
3442 GetInceptionFwdTest(input_size_, filter_size_, stride_,
3447 input_size_, filter_size_, stride_, padding_)))
3450 GetInceptionBackInputTest(input_size_, filter_size_,
3454 GetInceptionBackFilterTest(input_size_, filter_size_,
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc155 input_size_(input_size), in CudnnRnnParameters()
184 std::to_string(input_size_), in ToString()
202 return std::make_tuple(num_layers_, input_size_, num_units_, seq_length_, in get_data_as_tuple()
208 const int input_size_; member in tensorflow::__anond88b5f210111::CudnnRnnParameters
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1072 input_size_(input_size), in CudnnRnnDescriptor()
1204 int input_size() const { return input_size_; } in input_size()
1233 int input_size_; member in stream_executor::gpu::CudnnRnnDescriptor

12