Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsvdf.cc41 int scratch_tensor_index; member
79 int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare() local
H A Dunidirectional_sequence_gru.cc122 auto* scratch_tensor_index = new int; in Init() local
132 int* scratch_tensor_index = reinterpret_cast<int*>(node->user_data); in Prepare() local
H A Dadd_n.cc36 int scratch_tensor_index; member
H A Dunidirectional_sequence_lstm.cc43 int scratch_tensor_index; member
860 const int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare() local
H A Dbasic_rnn.cc32 int scratch_tensor_index; member
H A Dunidirectional_sequence_rnn.cc32 int scratch_tensor_index; member
H A Dtranspose_conv.cc74 int32_t scratch_tensor_index; member
H A Dbatch_matmul.cc65 int scratch_tensor_index; member
H A Dbidirectional_sequence_rnn.cc34 int scratch_tensor_index; member
H A Dbidirectional_sequence_lstm.cc158 int scratch_tensor_index; member
H A Dreduce.cc56 int scratch_tensor_index; member
H A Dfully_connected.cc120 int scratch_tensor_index; member
H A Dlstm.cc55 int scratch_tensor_index; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc537 int scratch_tensor_index; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc459 int scratch_tensor_index; member