Home
last modified time | relevance | path

Searched refs:use_projection (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/
Doperator_property.cc42 op_variant.use_projection = op->inputs[projection_weights_index] != -1; in GetOperatorVariant()
269 if (op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()
340 if (op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()
406 if (op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()
477 if (op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()
542 if (!op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()
612 if (!op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()
676 if (!op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()
745 if (!op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()
Doperator_property.h134 bool use_projection = false; member
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc274 const bool use_projection = (projection_weights != nullptr); in CalculateLstmOutputFloat() local
277 if (use_projection) { in CalculateLstmOutputFloat()
446 const bool use_projection = (projection_weights != nullptr); in CalculateLstmOutputHybrid() local
449 if (use_projection) { in CalculateLstmOutputHybrid()
631 const bool use_projection = (projection_weights != nullptr); in CalculateLstmOutputInteger8x8_16() local
633 if (use_projection) { in CalculateLstmOutputInteger8x8_16()
1402 const bool use_projection = (projection_weight_ptr != nullptr); in LstmStepInteger8x8_16() local
1403 if (use_projection) { in LstmStepInteger8x8_16()
Dlstm.cc238 const bool use_projection = (projection_weights != nullptr); in PopulateQuantizedLstmParams8x8_16() local
327 if (use_projection) { in PopulateQuantizedLstmParams8x8_16()
543 const bool use_projection = (projection_weights != nullptr); in PopulateQuantizedLstmParams8x8_8() local
642 if (use_projection) { in PopulateQuantizedLstmParams8x8_8()
Dunidirectional_sequence_lstm.cc153 const bool use_projection = (projection_weights != nullptr); in PopulateQuantizedLstmParams8x8_16() local
242 if (use_projection) { in PopulateQuantizedLstmParams8x8_16()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h98 lstm_variant->use_projection = in GetLstmProperty()
443 if (index == 4 && !lstm_variant.use_projection) { in processIntermediates()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc137 const bool use_projection = (projection_weights != nullptr); in CalculateLstmOutputCalibration() local
140 if (use_projection) { in CalculateLstmOutputCalibration()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2529 bool use_projection = cell_size != 0 && hidden_size < cell_size; in createRnnDescriptor() local
2530 if (use_projection) { in createRnnDescriptor()