Home
last modified time | relevance | path

Searched defs:n_cell (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc35 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
268 const int n_cell = 4; in TEST_P() local
342 const int n_cell = 4; in TEST_P() local
409 const int n_cell = 20; in TEST_P() local
972 const int n_cell = 4; in TEST_F() local
1058 const int n_cell = 4; in TEST_F() local
1122 const int n_cell = 4; in TEST_F() local
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm_test.cc36 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel()
251 const int n_cell = 4; in TEST_F() local
313 const int n_cell = 4; in TEST_F() local
380 const int n_cell = 4; in TEST_P() local
443 const int n_cell = 4; in TEST_P() local
554 const int n_cell = 4; in TEST_F() local
616 const int n_cell = 4; in TEST_P() local
678 const int n_cell = 4; in TEST_P() local
1340 const int n_cell = 20; in TEST_F() local
1407 const int n_cell = 20; in TEST_P() local
[all …]
Dbidirectional_sequence_lstm_test.cc32 BidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in BidirectionalLSTMOpModel()
433 const int n_cell = 4; in TEST_P() local
599 const int n_cell = 4; in TEST_P() local
765 const int n_cell = 4; in TEST() local
926 const int n_cell = 4; in TEST() local
1077 const int n_cell = 4; in TEST() local
1228 const int n_cell = 20; in TEST() local
1931 const int n_cell = 20; in TEST() local
2633 const int n_cell = 4; in TEST_P() local
2807 const int n_cell = 4; in TEST_P() local
Dlstm_eval.cc79 int32_t* projection_weights_row_sums, int32_t* row_sums, int n_cell, in ComputeRowSums()
188 const int n_output, const int n_cell, in CalculateLstmGateFloat()
255 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat()
302 void CalculateLstmOutputFloat(int n_batch, int n_cell, int n_output, in CalculateLstmOutputFloat()
340 void CalculateLstmGateHybrid( in CalculateLstmGateHybrid()
475 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputHybrid()
548 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_16()
613 void UpdateLstmCellInteger(int n_batch, int n_cell, int16_t* cell_state, in UpdateLstmCellInteger()
659 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_16()
713 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_8()
[all …]
Dlstm_test.cc39 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
416 const int n_cell = 4; in TEST_P() local
493 const int n_cell = 4; in TEST_P() local
563 const int n_cell = 20; in TEST_P() local
1146 const int n_cell = 4; in TEST_P() local
1241 const int n_cell = 4; in TEST_P() local
1314 LSTMIntegerOpModel(int n_batch, int n_input, int n_cell, int n_output, in LSTMIntegerOpModel()
1600 const int n_cell = 4; in TEST() local
1762 const int n_cell = 4; in TEST() local
1935 const int n_cell = 4; in TEST() local
[all …]
Doptional_tensor_test.cc31 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
232 const int n_cell = 4; in TEST() local
Dbidirectional_sequence_lstm.cc177 int n_cell, int input_to_input_weights_tensor, in CheckLstmTensorDimensionsAndTypes()
385 int n_output, int n_cell) { in CheckInputTensorDimensions()
839 int n_cell = std::max(n_fw_cell, n_bw_cell); in Prepare() local
Dunidirectional_sequence_lstm.cc419 int n_output, int n_cell, in CheckInputTensorDimensions()
903 const int n_cell = input_to_output_weights->dims->data[0]; in Prepare() local
Dlstm.cc855 int n_output, int n_cell, in CheckInputTensorDimensions()
1329 const int n_cell = input_to_output_weights->dims->data[0]; in Prepare() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc47 const int n_output, const int n_cell, in CalculateLstmGateFloat()
100 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat()
124 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputCalibration()
181 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepCalibration()
307 const int n_cell = input_to_output_weights->dims->data[0]; in EvalCalibration() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc63 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput()
384 const int n_cell = input_to_output_weights->dims->data[0]; in EvalFloat() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.h129 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
321 int32_t n_cell, int16_t* output) { in TwoGateSaturatingAdd()
Dsse_tensor_utils.h143 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
335 int32_t n_cell, int16_t* output) { in TwoGateSaturatingAdd()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h168 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
323 int32_t n_cell, int16_t* output) { in TwoGateSaturatingAdd()
Dportable_tensor_utils.cc383 int32_t n_cell, int8_t* gate_output, in PortableMatrixBatchVectorMultiply()
785 int32_t n_batch, int32_t n_cell, in PortableTwoGateSaturatingAdd()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3169 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
3535 const int n_cell = 4; in TEST_F() local
3599 const int n_cell = 4; in TEST_F() local
3711 const int n_cell = 4; in TEST_F() local
4369 const int n_cell = 20; in TEST_F() local
4502 const int n_cell = 4; in TEST_F() local
4643 const int n_cell = 4; in TEST_F() local
/external/armnn/src/backends/backendsCommon/
DWorkloadData.cpp2014 const uint32_t n_cell = m_InputToOutputWeights->GetShape()[0]; in Validate() local
3880 const uint32_t n_cell = m_InputToOutputWeights->GetShape()[0]; in Validate() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2521 const int n_cell = input_to_output_weights.getDimSize(0); in verify() local