Home
last modified time | relevance | path

Searched defs:n_cell (Results 1 – 17 of 17) 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.cc32 UnidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in UnidirectionalLSTMOpModel()
288 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel()
503 const int n_cell = 4; in TEST_F() local
565 const int n_cell = 4; in TEST_F() local
632 const int n_cell = 4; in TEST_P() local
695 const int n_cell = 4; in TEST_P() local
806 const int n_cell = 4; in TEST_F() local
868 const int n_cell = 4; in TEST_P() local
930 const int n_cell = 4; in TEST_P() local
1592 const int n_cell = 20; in TEST_F() 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.cc48 int32_t* projection_weights_row_sums, int32_t* row_sums, int n_cell, in ComputeRowSums()
157 const int n_output, const int n_cell, in CalculateLstmGateFloat()
218 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat()
262 void CalculateLstmOutputFloat(int n_batch, int n_cell, int n_output, in CalculateLstmOutputFloat()
298 void CalculateLstmGateHybrid( in CalculateLstmGateHybrid()
433 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputHybrid()
506 const int n_batch, const int n_input, const int n_output, const int n_cell, in CalculateLstmGateInteger8x8_16()
571 void UpdateLstmCellInteger(int n_batch, int n_cell, int16_t* cell_state, in UpdateLstmCellInteger()
617 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_16()
671 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()
835 int n_cell = std::max(n_fw_cell, n_bw_cell); in Prepare() local
Dunidirectional_sequence_lstm.cc411 int n_output, int n_cell, in CheckInputTensorDimensions()
893 const int n_cell = input_to_output_weights->dims->data[0]; in Prepare() local
Dlstm.cc854 int n_output, int n_cell, in CheckInputTensorDimensions()
1328 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()
180 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepCalibration()
306 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()
375 const int n_cell = input_to_output_weights->dims->data[0]; in EvalFloat() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h153 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
309 int32_t n_cell, int16_t* output) { in TwoGateSaturatingAdd()
Dportable_tensor_utils.cc348 int32_t n_cell, int8_t* gate_output, in PortableMatrixBatchVectorMultiply()
750 int32_t n_batch, int32_t n_cell, in PortableTwoGateSaturatingAdd()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.h114 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
306 int32_t n_cell, int16_t* output) { in TwoGateSaturatingAdd()
Dsse_tensor_utils.h123 int32_t n_batch, int32_t n_input, int32_t n_cell, in MatrixBatchVectorMultiply()
315 int32_t n_cell, int16_t* output) { in TwoGateSaturatingAdd()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3092 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
3458 const int n_cell = 4; in TEST_F() local
3522 const int n_cell = 4; in TEST_F() local
3634 const int n_cell = 4; in TEST_F() local
4292 const int n_cell = 20; in TEST_F() local
4425 const int n_cell = 4; in TEST_F() local
4566 const int n_cell = 4; in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2028 const int n_cell = input_to_output_weights.getDimSize(0); in Verify() local