Searched refs:activation_state (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | svdf.cc | 122 TfLiteEvalTensor* activation_state, TfLiteEvalTensor* output) { in EvalFloatSVDF() argument
137 float* state_ptr = tflite::micro::GetTensorData<float>(activation_state); in EvalFloatSVDF()
216 const TfLiteTensor* activation_state = in Prepare() local
218 TF_LITE_ENSURE(context, activation_state != nullptr); in Prepare()
258 TF_LITE_ENSURE_EQ(context, NumDimensions(activation_state), 2); in Prepare()
259 TF_LITE_ENSURE_EQ(context, activation_state->dims->data[0], batch_size); in Prepare()
260 TF_LITE_ENSURE_EQ(context, activation_state->dims->data[1], in Prepare()
263 TF_LITE_ENSURE_EQ(context, activation_state->is_variable, true); in Prepare()
273 TF_LITE_ENSURE_EQ(context, activation_state->type, kTfLiteInt16); in Prepare()
282 activation_state->params.scale); in Prepare()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | svdf.cc | 138 TfLiteEvalTensor* activation_state, TfLiteEvalTensor* output) { in EvalFloatSVDF() argument
153 float* state_ptr = tflite::micro::GetTensorData<float>(activation_state); in EvalFloatSVDF()
304 const TfLiteTensor* activation_state = in Prepare() local
344 TF_LITE_ENSURE_EQ(context, NumDimensions(activation_state), 2); in Prepare()
345 TF_LITE_ENSURE_EQ(context, activation_state->dims->data[0], batch_size); in Prepare()
346 TF_LITE_ENSURE_EQ(context, activation_state->dims->data[1], in Prepare()
349 TF_LITE_ENSURE_EQ(context, activation_state->is_variable, true); in Prepare()
359 TF_LITE_ENSURE_EQ(context, activation_state->type, kTfLiteInt16); in Prepare()
368 activation_state->params.scale); in Prepare()
370 static_cast<double>(activation_state->params.scale * in Prepare()
[all …]
|
| /external/autotest/client/cros/cellular/pseudomodem/ |
| D | modem_cdma.py | 76 activation_state = \
79 activation_state = \
92 'ActivationState' : dbus.types.UInt32(activation_state)
186 activation_state, argument
202 activation_state,
271 activation_state = \
274 activation_state = \
280 activation_state = \
283 activation_state)
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | svdf.cc | 339 const TfLiteTensor* activation_state = in Prepare() local
394 TF_LITE_ENSURE_EQ(context, NumDimensions(activation_state), 2); in Prepare()
395 TF_LITE_ENSURE_EQ(context, activation_state->dims->data[0], batch_size); in Prepare()
396 TF_LITE_ENSURE_EQ(context, activation_state->dims->data[1], in Prepare()
402 TF_LITE_ENSURE_EQ(context, activation_state->type, kTfLiteInt16); in Prepare()
409 activation_state->params.scale); in Prepare()
411 static_cast<double>(activation_state->params.scale * in Prepare()
415 static_cast<double>(activation_state->params.scale * in Prepare()
463 TfLiteEvalTensor* activation_state = tflite::micro::GetMutableEvalInput( in Eval() local
473 bias, params, activation_state, output, data); in Eval()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 357 TfLiteTensor* activation_state, TfLiteTensor* cell_state, in EvalFloat()
444 GetTensorData<float>(activation_state), in EvalFloat()
469 float* activation_state_ptr = GetTensorData<float>(activation_state) + in EvalFloat()
594 TfLiteTensor* activation_state = GetVariableInput( in lstm_eval() local
596 TF_LITE_ENSURE(context, activation_state != nullptr); in lstm_eval()
627 /*output_offset=*/0, scratch_buffer, activation_state, cell_state, in lstm_eval()
|
| /external/autotest/client/cros/cellular/mbim_compliance/tests/ |
| D | cm_13.py | 44 activation_state=0,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2017 const auto activation_state = in Verify() local
2024 if (activation_state.hasStaticShape() && cell_state.hasStaticShape() && in Verify()
2030 const int output_state_size = activation_state.getNumElements(); in Verify()
|
| D | tfl_ops.td | 4349 TFL_StatefulTensor:$activation_state,
|