Home
last modified time | relevance | path

Searched refs:scratch_tensor_index (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dsvdf.cc41 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
146 node->temporaries->data[0] = scratch_tensor_index; in Prepare()
171 node->temporaries->data[1] = scratch_tensor_index + 1; in Prepare()
184 node->temporaries->data[2] = scratch_tensor_index + 2; in Prepare()
200 node->temporaries->data[3] = scratch_tensor_index + 3; in Prepare()
215 node->temporaries->data[4] = scratch_tensor_index + 4; in Prepare()
229 node->temporaries->data[5] = scratch_tensor_index + 5; in Prepare()
248 node->temporaries->data[1] = scratch_tensor_index + 1; in Prepare()
Dbasic_rnn.cc32 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()
Dunidirectional_sequence_rnn.cc32 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()
Dbidirectional_sequence_lstm.cc158 int scratch_tensor_index; member
166 &op_data->scratch_tensor_index); in Init()
570 op_data->scratch_tensor_index + kFwScratchBuffer; in Prepare()
632 op_data->scratch_tensor_index + kBwScratchBuffer; in Prepare()
664 op_data->scratch_tensor_index + kInputQuantized; in Prepare()
677 op_data->scratch_tensor_index + kFwActivationStateQuantized; in Prepare()
693 op_data->scratch_tensor_index + kBwActivationStateQuantized; in Prepare()
709 op_data->scratch_tensor_index + kFwCellStateQuantized; in Prepare()
725 op_data->scratch_tensor_index + kBwCellStateQuantized; in Prepare()
747 op_data->scratch_tensor_index + kInputScalingFactors; in Prepare()
[all …]
Dunidirectional_sequence_lstm.cc42 int scratch_tensor_index; member
400 &op_data->scratch_tensor_index); in Init()
850 const int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare() local
950 scratch_tensor_index + kScratchBuffer; in Prepare()
979 scratch_tensor_index + kInputQuantized; in Prepare()
991 scratch_tensor_index + kOutputStateQuantized; in Prepare()
1007 scratch_tensor_index + kCellStateQuantized; in Prepare()
1028 op_data->scratch_tensor_index + kInputScalingFactors; in Prepare()
1043 op_data->scratch_tensor_index + kOutputStateScalingFactors; in Prepare()
1057 scratch_tensor_index + kProductScalingFactors; in Prepare()
[all …]
Dbidirectional_sequence_rnn.cc34 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()
312 op_data->scratch_tensor_index + kBwRowSums; in Prepare()
[all …]
Dbatch_matmul.cc65 int scratch_tensor_index; member
91 &op_data->scratch_tensor_index); in Init()
154 node->temporaries->data[0] = op_data->scratch_tensor_index; in InitializeTemporaries()
177 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in InitializeTemporaries()
218 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in InitializeTemporaries()
230 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in InitializeTemporaries()
245 node->temporaries->data[4] = op_data->scratch_tensor_index + 4; in InitializeTemporaries()
261 node->temporaries->data[5] = op_data->scratch_tensor_index + 5; in InitializeTemporaries()
273 node->temporaries->data[6] = op_data->scratch_tensor_index + 6; in InitializeTemporaries()
Dtranspose_conv.cc74 int32_t scratch_tensor_index; member
157 data->scratch_tensor_index = temporaries_count; in AllocateTemporaryTensorsIfRequired()
350 node->temporaries->data[data->scratch_tensor_index] = in Prepare()
354 context, GetTemporarySafe(context, node, data->scratch_tensor_index, in Prepare()
631 context, GetTemporarySafe(context, node, data->scratch_tensor_index, in Eval()
650 context, GetTemporarySafe(context, node, data->scratch_tensor_index, in Eval()
667 context, GetTemporarySafe(context, node, data->scratch_tensor_index, in Eval()
Dfully_connected.cc116 int scratch_tensor_index; member
181 &op_data->scratch_tensor_index); in Init()
270 node->temporaries->data[0] = data->scratch_tensor_index; in PrepareImpl()
282 node->temporaries->data[1] = data->scratch_tensor_index + 1; in PrepareImpl()
297 node->temporaries->data[2] = data->scratch_tensor_index + 2; in PrepareImpl()
313 node->temporaries->data[3] = data->scratch_tensor_index + 3; in PrepareImpl()
325 node->temporaries->data[4] = data->scratch_tensor_index + 4; in PrepareImpl()
341 node->temporaries->data[5] = data->scratch_tensor_index + 5; in PrepareImpl()
Dlstm.cc55 int scratch_tensor_index; member
843 &op_data->scratch_tensor_index); in Init()
1408 op_data->scratch_tensor_index + kScratchBuffer; in Prepare()
1438 op_data->scratch_tensor_index + kInputQuantized; in Prepare()
1450 op_data->scratch_tensor_index + kOutputStateQuantized; in Prepare()
1466 op_data->scratch_tensor_index + kCellStateQuantized; in Prepare()
1486 op_data->scratch_tensor_index + kInputScalingFactors; in Prepare()
1501 op_data->scratch_tensor_index + kOutputStateScalingFactors; in Prepare()
1515 op_data->scratch_tensor_index + kProductScalingFactors; in Prepare()
1534 op_data->scratch_tensor_index + kRecoveredCellWeights; in Prepare()
[all …]
Dreduce.cc53 int scratch_tensor_index; member
73 context->AddTensors(context, 3, &op_data->scratch_tensor_index); in Init()
172 node->temporaries->data[0] = op_data->scratch_tensor_index; in InitializeTemporaries()
184 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in InitializeTemporaries()
190 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in InitializeTemporaries()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc114 auto* scratch_tensor_index = new int; in Init() local
115 context->AddTensors(context, kTemporaryNum, scratch_tensor_index); in Init()
116 return scratch_tensor_index; in Init()
124 int* scratch_tensor_index = reinterpret_cast<int*>(node->user_data); in Prepare() local
198 node->temporaries->data[kActivation] = *scratch_tensor_index; in Prepare()
211 node->temporaries->data[kConcat] = (*scratch_tensor_index) + kConcat; in Prepare()
/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() argument
159 context->GetScratchBuffer(context, scratch_tensor_index)); in EvalFloatSVDF()
261 context->GetScratchBuffer(context, data.scratch_tensor_index)); in EvalIntegerSVDF()
392 &(data->scratch_tensor_index)); in Prepare()
412 &(data->scratch_tensor_index)); in Prepare()
442 params, data.scratch_tensor_index, activation_state, in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dsvdf.cc121 const TfLiteSVDFParams* params, int scratch_tensor_index, in EvalFloatSVDF() argument
143 context->GetScratchBuffer(context, scratch_tensor_index)); in EvalFloatSVDF()
306 &(data->scratch_tensor_index)); in Prepare()
326 &(data->scratch_tensor_index)); in Prepare()
356 params, data.scratch_tensor_index, activation_state, in Eval()
Dsvdf.h30 int scratch_tensor_index; member
Dsvdf_common.cc52 context->GetScratchBuffer(context, data.scratch_tensor_index)); in EvalIntegerSvdfReference()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsoftmax.cc39 int scratch_tensor_index;
206 context, required_scratch, &(data->scratch_tensor_index))); in PrepareHifi4()
225 context->GetScratchBuffer(context, op_data->scratch_tensor_index)); in EvalHifi4()
Dsvdf.cc73 context->GetScratchBuffer(context, data.scratch_tensor_index)); in EvalIntegerSvdfHifimini()
439 &(data->scratch_tensor_index)); in Prepare()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc458 int scratch_tensor_index; member
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc526 int scratch_tensor_index; member