Home
last modified time | relevance | path

Searched defs:scratch_tensor_index (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dsvdf.h30 int scratch_tensor_index; member
Dsvdf.cc121 const TfLiteSVDFParams* params, int scratch_tensor_index, in EvalFloatSVDF()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc114 auto* scratch_tensor_index = new int; in Init() local
124 int* scratch_tensor_index = reinterpret_cast<int*>(node->user_data); in Prepare() local
/external/tensorflow/tensorflow/lite/kernels/
Dsvdf.cc41 int scratch_tensor_index; member
79 int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare() local
Dbasic_rnn.cc32 int scratch_tensor_index; member
Dunidirectional_sequence_lstm.cc42 int scratch_tensor_index; member
850 const int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare() local
Dunidirectional_sequence_rnn.cc32 int scratch_tensor_index; member
Dtranspose_conv.cc74 int32_t scratch_tensor_index; member
Dreduce.cc53 int scratch_tensor_index; member
Dbatch_matmul.cc65 int scratch_tensor_index; member
Dbidirectional_sequence_rnn.cc34 int scratch_tensor_index; member
Dfully_connected.cc116 int scratch_tensor_index; member
Dbidirectional_sequence_lstm.cc158 int scratch_tensor_index; member
Dlstm.cc55 int scratch_tensor_index; member
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsvdf.cc42 int scratch_tensor_index; member
137 const TfLiteSVDFParams* params, int scratch_tensor_index, in EvalFloatSVDF()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc526 int scratch_tensor_index; member
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc458 int scratch_tensor_index; member