Home
last modified time | relevance | path

Searched defs:n_output (Results 1 – 25 of 26) sorted by relevance

12

/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()
269 const int n_output = 4; in TEST_P() local
343 const int n_output = 4; in TEST_P() local
410 const int n_output = 16; in TEST_P() local
973 const int n_output = 3; in TEST_F() local
1059 const int n_output = 3; in TEST_F() local
1123 const int n_output = 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()
504 const int n_output = 4; in TEST_F() local
566 const int n_output = 4; in TEST_F() local
633 const int n_output = 4; in TEST_P() local
696 const int n_output = 4; in TEST_P() local
807 const int n_output = 4; in TEST_F() local
869 const int n_output = 4; in TEST_P() local
931 const int n_output = 4; in TEST_P() local
1593 const int n_output = 16; 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()
434 const int n_output = 4; in TEST_P() local
600 const int n_output = 4; in TEST_P() local
766 const int n_output = 4; in TEST() local
927 const int n_output = 4; in TEST() local
1078 const int n_output = 4; in TEST() local
1229 const int n_output = 16; in TEST() local
1932 const int n_output = 16; in TEST() local
2634 const int n_output = 4; in TEST_P() local
2808 const int n_output = 4; in TEST_P() local
Dlstm_test.cc39 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
417 const int n_output = 4; in TEST_P() local
494 const int n_output = 4; in TEST_P() local
564 const int n_output = 16; in TEST_P() local
1147 const int n_output = 3; in TEST_P() local
1242 const int n_output = 3; in TEST_P() local
1314 LSTMIntegerOpModel(int n_batch, int n_input, int n_cell, int n_output, in LSTMIntegerOpModel()
1601 const int n_output = 3; in TEST() local
1763 const int n_output = 3; in TEST() local
1936 const int n_output = 3; in TEST() local
[all …]
Dlstm_eval.cc49 int n_input, int n_aux_input, int n_output, in ComputeRowSums()
157 const int n_output, const int n_cell, in CalculateLstmGateFloat()
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()
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()
725 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_8()
808 inline void LstmStepFloat( in LstmStepFloat()
[all …]
Doptional_tensor_test.cc31 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
233 const int n_output = 4; in TEST() local
Dunidirectional_sequence_lstm.cc411 int n_output, int n_cell, in CheckInputTensorDimensions()
905 const int n_output = recurrent_to_output_weights->dims->data[1]; in Prepare() local
Dbidirectional_sequence_lstm.cc176 TfLiteContext* context, TfLiteNode* node, int n_input, int n_output, in CheckLstmTensorDimensionsAndTypes()
385 int n_output, int n_cell) { in CheckInputTensorDimensions()
Dlstm.cc854 int n_output, int n_cell, in CheckInputTensorDimensions()
1339 const int n_output = recurrent_to_output_weights->dims->data[1]; in Prepare() local
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru_test.cc34 explicit GRUOpModel(int n_batch, int n_input, int n_output, in GRUOpModel()
103 const int n_output = 3; in TEST() local
Dunidirectional_sequence_gru.cc40 const int n_output = output->dims->data[2]; in GruImpl() local
143 const int n_output = input_state->dims->data[1]; in Prepare() local
Dgru_cell.cc47 const int n_output = state_shape.Dims(1); in GruCell() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc47 const int n_output, const int n_cell, in CalculateLstmGateFloat()
124 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputCalibration()
157 inline void LstmStepCalibration( in LstmStepCalibration()
307 const int n_output = recurrent_to_output_weights->dims->data[1]; in EvalCalibration() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h126 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
136 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
166 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
Dportable_tensor_utils.cc296 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulateImpl()
325 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulate()
335 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulate()
378 int32_t n_output, int32_t output_zp, int8_t* proj_output) { in PortableMatrixBatchVectorMultiply()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.h102 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
112 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
136 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
Dneon_tensor_utils.h93 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
103 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate()
127 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
Dneon_tensor_utils.cc744 int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyImpl()
853 int32_t multiplier, int32_t shift, int32_t n_batch, int32_t n_output, in NeonMatrixBatchVectorAccumulateImpl()
906 int32_t multiplier, int32_t shift, int32_t n_batch, int32_t n_output, in NeonMatrixBatchVectorAccumulateImpl()
981 int32_t n_input, int32_t n_output, int32_t output_zp, in NeonCpuBackendGemm()
1014 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyAccumulate()
1030 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyAccumulate()
Dsse_tensor_utils.cc178 int32_t n_input, int32_t n_output, int32_t output_zp, in SseCpuBackendGemm()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
376 const int n_output = recurrent_to_output_weights->dims->data[1]; in EvalFloat() local
/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()
3459 const int n_output = 4; in TEST_F() local
3523 const int n_output = 4; in TEST_F() local
3635 const int n_output = 4; in TEST_F() local
4293 const int n_output = 16; in TEST_F() local
4426 const int n_output = 3; in TEST_F() local
4567 const int n_output = 3; in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc682 const int n_output = recurrent_kernel_type.getDimSize(0); in ConvertKerasLSTMLayer() local
/external/deqp/external/openglcts/modules/glesext/geometry_shader/
DesextcGeometryShaderRendering.cpp425 unsigned int n_output = 0; in init() local
/external/libxml2/
Dtestapi.c6983 int n_output; in test_xmlDebugCheckDocument() local
7023 int n_output; in test_xmlDebugDumpAttr() local
7069 int n_output; in test_xmlDebugDumpAttrList() local
7115 int n_output; in test_xmlDebugDumpDTD() local
7154 int n_output; in test_xmlDebugDumpDocument() local
7193 int n_output; in test_xmlDebugDumpDocumentHead() local
7232 int n_output; in test_xmlDebugDumpEntities() local
7271 int n_output; in test_xmlDebugDumpNode() local
7317 int n_output; in test_xmlDebugDumpNodeList() local
7363 int n_output; in test_xmlDebugDumpOneNode() local
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2029 const int n_output = recurrent_to_output_weights.getDimSize(1); in Verify() local

12