| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | svdf.cc | 41 int scratch_tensor_index; member
68 &op_data->scratch_tensor_index); in Init()
79 int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare() local
147 node->temporaries->data[0] = scratch_tensor_index; in Prepare()
172 node->temporaries->data[1] = scratch_tensor_index + 1; in Prepare()
185 node->temporaries->data[2] = scratch_tensor_index + 2; in Prepare()
201 node->temporaries->data[3] = scratch_tensor_index + 3; in Prepare()
217 node->temporaries->data[4] = scratch_tensor_index + 4; in Prepare()
231 node->temporaries->data[5] = scratch_tensor_index + 5; in Prepare()
251 node->temporaries->data[1] = scratch_tensor_index + 1; in Prepare()
|
| H A D | bidirectional_sequence_lstm.cc | 158 int scratch_tensor_index; member
166 &op_data->scratch_tensor_index); in Init()
570 op_data->scratch_tensor_index + kFwScratchBuffer; in Prepare()
634 op_data->scratch_tensor_index + kBwScratchBuffer; in Prepare()
668 op_data->scratch_tensor_index + kInputQuantized; in Prepare()
681 op_data->scratch_tensor_index + kFwActivationStateQuantized; in Prepare()
697 op_data->scratch_tensor_index + kBwActivationStateQuantized; in Prepare()
713 op_data->scratch_tensor_index + kFwCellStateQuantized; in Prepare()
729 op_data->scratch_tensor_index + kBwCellStateQuantized; in Prepare()
751 op_data->scratch_tensor_index + kInputScalingFactors; in Prepare()
[all …]
|
| H A D | basic_rnn.cc | 32 int scratch_tensor_index; member
50 &op_data->scratch_tensor_index); in Init()
116 node->temporaries->data[0] = op_data->scratch_tensor_index; in Prepare()
127 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in Prepare()
141 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in Prepare()
154 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in Prepare()
169 node->temporaries->data[4] = op_data->scratch_tensor_index + 4; in Prepare()
182 node->temporaries->data[5] = op_data->scratch_tensor_index + 5; in Prepare()
|
| H A D | unidirectional_sequence_rnn.cc | 32 int scratch_tensor_index; member
51 &op_data->scratch_tensor_index); in Init()
123 node->temporaries->data[0] = op_data->scratch_tensor_index; in Prepare()
134 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in Prepare()
148 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in Prepare()
161 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in Prepare()
176 node->temporaries->data[4] = op_data->scratch_tensor_index + 4; in Prepare()
189 node->temporaries->data[5] = op_data->scratch_tensor_index + 5; in Prepare()
|
| H A D | unidirectional_sequence_gru.cc | 122 auto* scratch_tensor_index = new int; in Init() local
123 context->AddTensors(context, kTemporaryNum, scratch_tensor_index); in Init()
124 return scratch_tensor_index; in Init()
132 int* scratch_tensor_index = reinterpret_cast<int*>(node->user_data); in Prepare() local
206 node->temporaries->data[kActivation] = *scratch_tensor_index; in Prepare()
219 node->temporaries->data[kConcat] = (*scratch_tensor_index) + kConcat; in Prepare()
|
| H A D | unidirectional_sequence_lstm.cc | 43 int scratch_tensor_index; member
408 &op_data->scratch_tensor_index); in Init()
860 const int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare() local
960 scratch_tensor_index + kScratchBuffer; in Prepare()
991 scratch_tensor_index + kInputQuantized; in Prepare()
1003 scratch_tensor_index + kOutputStateQuantized; in Prepare()
1019 scratch_tensor_index + kCellStateQuantized; in Prepare()
1040 op_data->scratch_tensor_index + kInputScalingFactors; in Prepare()
1055 op_data->scratch_tensor_index + kOutputStateScalingFactors; in Prepare()
1069 scratch_tensor_index + kProductScalingFactors; in Prepare()
[all …]
|
| H A D | bidirectional_sequence_rnn.cc | 34 int scratch_tensor_index; member
83 &op_data->scratch_tensor_index); in Init()
201 op_data->scratch_tensor_index + kInputQuantized; in Prepare()
214 op_data->scratch_tensor_index + kFwHiddenStateQuantized; in Prepare()
231 op_data->scratch_tensor_index + kBwHiddenStateQuantized; in Prepare()
249 op_data->scratch_tensor_index + kScalingFactors; in Prepare()
263 op_data->scratch_tensor_index + kAccumScratch; in Prepare()
280 op_data->scratch_tensor_index + kZeroPoints; in Prepare()
296 op_data->scratch_tensor_index + kFwRowSums; in Prepare()
313 op_data->scratch_tensor_index + kBwRowSums; in Prepare()
[all …]
|
| H A D | batch_matmul.cc | 65 int scratch_tensor_index; member
91 &op_data->scratch_tensor_index); in Init()
156 node->temporaries->data[0] = op_data->scratch_tensor_index; in InitializeTemporaries()
179 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in InitializeTemporaries()
221 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in InitializeTemporaries()
233 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in InitializeTemporaries()
248 node->temporaries->data[4] = op_data->scratch_tensor_index + 4; in InitializeTemporaries()
264 node->temporaries->data[5] = op_data->scratch_tensor_index + 5; in InitializeTemporaries()
276 node->temporaries->data[6] = op_data->scratch_tensor_index + 6; in InitializeTemporaries()
|
| H A D | add_n.cc | 36 int scratch_tensor_index; member
41 context->AddTensors(context, 1, &op_data->scratch_tensor_index); in Init()
65 node->temporaries->data[0] = op_data->scratch_tensor_index; in Prepare()
|
| H A D | transpose_conv.cc | 74 int32_t scratch_tensor_index; member
157 data->scratch_tensor_index = temporaries_count; in AllocateTemporaryTensorsIfRequired()
351 node->temporaries->data[data->scratch_tensor_index] = in Prepare()
355 context, GetTemporarySafe(context, node, data->scratch_tensor_index, in Prepare()
665 context, GetTemporarySafe(context, node, data->scratch_tensor_index, in Eval()
684 context, GetTemporarySafe(context, node, data->scratch_tensor_index, in Eval()
701 context, GetTemporarySafe(context, node, data->scratch_tensor_index, in Eval()
|
| H A D | lstm.cc | 55 int scratch_tensor_index; member
844 &op_data->scratch_tensor_index); in Init()
1409 op_data->scratch_tensor_index + kScratchBuffer; in Prepare()
1441 op_data->scratch_tensor_index + kInputQuantized; in Prepare()
1453 op_data->scratch_tensor_index + kOutputStateQuantized; in Prepare()
1469 op_data->scratch_tensor_index + kCellStateQuantized; in Prepare()
1489 op_data->scratch_tensor_index + kInputScalingFactors; in Prepare()
1504 op_data->scratch_tensor_index + kOutputStateScalingFactors; in Prepare()
1518 op_data->scratch_tensor_index + kProductScalingFactors; in Prepare()
1537 op_data->scratch_tensor_index + kRecoveredCellWeights; in Prepare()
[all …]
|
| H A D | fully_connected.cc | 120 int scratch_tensor_index; member
185 &op_data->scratch_tensor_index); in Init()
347 node->temporaries->data[0] = data->scratch_tensor_index; in PrepareImpl()
359 node->temporaries->data[1] = data->scratch_tensor_index + 1; in PrepareImpl()
374 node->temporaries->data[2] = data->scratch_tensor_index + 2; in PrepareImpl()
390 node->temporaries->data[3] = data->scratch_tensor_index + 3; in PrepareImpl()
402 node->temporaries->data[4] = data->scratch_tensor_index + 4; in PrepareImpl()
418 node->temporaries->data[5] = data->scratch_tensor_index + 5; in PrepareImpl()
|
| H A D | reduce.cc | 56 int scratch_tensor_index; member
76 context->AddTensors(context, 4, &op_data->scratch_tensor_index); in Init()
183 node->temporaries->data[0] = op_data->scratch_tensor_index; in InitializeTemporaries()
195 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in InitializeTemporaries()
202 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in InitializeTemporaries()
228 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in InitializeTemporaries()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 537 int scratch_tensor_index; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 459 int scratch_tensor_index; member
|