Home
last modified time | relevance | path

Searched refs:temporaries (Results 1 – 25 of 109) sorted by relevance

12345

/external/tensorflow/tensorflow/lite/kernels/
Dsvdf.cc134 TfLiteIntArrayFree(node->temporaries); in Prepare()
136 node->temporaries = TfLiteIntArrayCreate(4); in Prepare()
138 node->temporaries = TfLiteIntArrayCreate(2); in Prepare()
140 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
142 node->temporaries->data[0] = scratch_tensor_index; in Prepare()
164 node->temporaries->data[1] = scratch_tensor_index + 1; in Prepare()
175 node->temporaries->data[2] = scratch_tensor_index + 2; in Prepare()
189 node->temporaries->data[3] = scratch_tensor_index + 3; in Prepare()
207 node->temporaries->data[1] = scratch_tensor_index + 1; in Prepare()
Dunidirectional_sequence_lstm.cc382 TfLiteIntArrayFree(node->temporaries); in Prepare()
384 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors); in Prepare()
386 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
388 node->temporaries->data[0] = scratch_tensor_index; in Prepare()
413 node->temporaries->data[kInputQuantized] = in Prepare()
424 node->temporaries->data[kOutputStateQuantized] = in Prepare()
438 node->temporaries->data[kCellStateQuantized] = in Prepare()
457 node->temporaries->data[kScalingFactors] = in Prepare()
470 node->temporaries->data[kProductScalingFactors] = in Prepare()
487 node->temporaries->data[kRecoveredCellWeights] = in Prepare()
[all …]
Dbidirectional_sequence_lstm.cc517 TfLiteIntArrayFree(node->temporaries); in Prepare()
519 node->temporaries = TfLiteIntArrayCreate( in Prepare()
522 node->temporaries = TfLiteIntArrayCreate(2); // the two scratch buffers. in Prepare()
525 node->temporaries->data[kFwScratchBuffer] = *scratch_tensor_index; in Prepare()
583 node->temporaries->data[kBwScratchBuffer] = in Prepare()
611 node->temporaries->data[kInputQuantized] = in Prepare()
623 node->temporaries->data[kFwActivationStateQuantized] = in Prepare()
637 node->temporaries->data[kBwActivationStateQuantized] = in Prepare()
651 node->temporaries->data[kFwCellStateQuantized] = in Prepare()
665 node->temporaries->data[kBwCellStateQuantized] = in Prepare()
[all …]
Dbasic_rnn.cc93 TfLiteIntArrayFree(node->temporaries); in Prepare()
94 node->temporaries = TfLiteIntArrayCreate(3); in Prepare()
95 node->temporaries->data[0] = *scratch_tensor_index; in Prepare()
104 node->temporaries->data[1] = *scratch_tensor_index + 1; in Prepare()
117 node->temporaries->data[2] = *scratch_tensor_index + 2; in Prepare()
Dnumeric_verify.cc91 TfLiteIntArrayFree(node->temporaries); in Prepare()
92 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
93 node->temporaries->data[0] = op_data->cache_tensor_id; in Prepare()
Dconv.cc284 TfLiteIntArrayFree(node->temporaries); in AllocateTemporaryTensorsIfRequired()
285 node->temporaries = TfLiteIntArrayCreate(temporaries_count); in AllocateTemporaryTensorsIfRequired()
419 node->temporaries->data[data->im2col_index] = data->im2col_id; in Prepare()
430 &context->tensors[node->temporaries->data[data->im2col_index]]; in Prepare()
441 node->temporaries->data[data->hwcn_weights_index] = data->hwcn_weights_id; in Prepare()
453 &context->tensors[node->temporaries->data[data->hwcn_weights_index]]; in Prepare()
467 node->temporaries->data[data->input_quantized_index] = in Prepare()
479 node->temporaries->data[data->scaling_factors_index] = in Prepare()
497 node->temporaries->data[data->accum_scratch_index] = data->accum_scratch_id; in Prepare()
519 node->temporaries->data[data->input_offset_index] = data->input_offset_id; in Prepare()
[all …]
Dunidirectional_sequence_rnn.cc100 TfLiteIntArrayFree(node->temporaries); in Prepare()
101 node->temporaries = TfLiteIntArrayCreate(3); in Prepare()
102 node->temporaries->data[0] = *scratch_tensor_index; in Prepare()
111 node->temporaries->data[1] = *scratch_tensor_index + 1; in Prepare()
124 node->temporaries->data[2] = *scratch_tensor_index + 2; in Prepare()
Dbidirectional_sequence_rnn.cc162 TfLiteIntArrayFree(node->temporaries); in Prepare()
164 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors); in Prepare()
167 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors - 1); in Prepare()
170 node->temporaries->data[kInputQuantized] = in Prepare()
182 node->temporaries->data[kFwHiddenStateQuantized] = in Prepare()
197 node->temporaries->data[kBwHiddenStateQuantized] = in Prepare()
213 node->temporaries->data[kScalingFactors] = in Prepare()
228 node->temporaries->data[kAuxInputQuantized] = in Prepare()
Drfft2d.cc57 TfLiteIntArrayFree(node->temporaries); in InitTemporaryTensors()
59 node->temporaries = TfLiteIntArrayCreate(2); in InitTemporaryTensors()
62 node->temporaries->data[kFftIntegerWorkingAreaTensor] = first_new_index; in InitTemporaryTensors()
64 node->temporaries->data[kFftDoubleWorkingAreaTensor] = first_new_index + 1; in InitTemporaryTensors()
Dlstm.cc870 TfLiteIntArrayFree(node->temporaries); in Prepare()
872 node->temporaries = TfLiteIntArrayCreate(8); in Prepare()
874 node->temporaries = TfLiteIntArrayCreate(6); in Prepare()
876 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
883 node->temporaries->data[0] = op_data->scratch_tensor_index; in Prepare()
907 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in Prepare()
916 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in Prepare()
929 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in Prepare()
946 node->temporaries->data[4] = op_data->scratch_tensor_index + 4; in Prepare()
957 node->temporaries->data[5] = op_data->scratch_tensor_index + 5; in Prepare()
[all …]
Dtranspose_conv.cc166 TfLiteIntArrayFree(node->temporaries); in AllocateTemporaryTensorsIfRequired()
167 node->temporaries = TfLiteIntArrayCreate(temporaries_count); in AllocateTemporaryTensorsIfRequired()
278 node->temporaries->data[data->col2im_index] = data->col2im_id; in Prepare()
297 node->temporaries->data[data->transposed_weights_index] = in Prepare()
309 node->temporaries->data[data->scratch_tensor_index] = in Prepare()
Ddepthwise_conv.cc213 TfLiteIntArrayFree(node->temporaries); in Prepare()
214 node->temporaries = TfLiteIntArrayCreate(temporaries_count); in Prepare()
216 node->temporaries->data[data->input_quantized_index] = in Prepare()
227 node->temporaries->data[data->scaling_factors_index] = in Prepare()
241 node->temporaries->data[data->input_offset_index] = data->input_offset_id; in Prepare()
/external/tensorflow/tensorflow/lite/
Doptional_debug_tools.cc122 if (node.temporaries && node.temporaries->size) { in PrintInterpreterState()
124 PrintTfLiteIntVector(node.temporaries); in PrintInterpreterState()
Darena_planner_test.cc34 std::initializer_list<int> temporaries) in TestOp() argument
35 : inputs_(inputs), outputs_(outputs), temporaries_(temporaries) {} in TestOp()
39 const std::vector<int>& temporaries() const { return temporaries_; } in temporaries() function in tflite::__anon6b6c45860111::TestOp
79 nodes_.back().temporaries = int_array(node.temporaries()); in TestGraph()
80 for (int t : node.temporaries()) { in TestGraph()
98 TfLiteIntArrayFree(node.temporaries); in ~TestGraph()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc181 TfLiteIntArrayFree(node->temporaries); in Prepare()
182 node->temporaries = TfLiteIntArrayCreate(kTemporaryNum); in Prepare()
185 node->temporaries->data[kActivation] = *scratch_tensor_index; in Prepare()
196 node->temporaries->data[kConcat] = (*scratch_tensor_index) + kConcat; in Prepare()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dactivations_test.cc63 node.temporaries = nullptr; in TestReluFloat()
119 node.temporaries = nullptr; in TestRelu6Float()
180 node.temporaries = nullptr; in TestReluUint8()
245 node.temporaries = nullptr; in TestRelu6Uint8()
309 node.temporaries = nullptr; in TestReluInt8()
375 node.temporaries = nullptr; in TestRelu6Int8()
Dpack_test.cc74 node.temporaries = temporaries_array; in TestPackTwoInputsFloat()
151 node.temporaries = temporaries_array; in TestPackThreeInputsFloat()
224 node.temporaries = temporaries_array; in TestPackTwoInputsQuantized()
294 node.temporaries = temporaries_array; in TestPackTwoInputsQuantized32()
Dmaximum_minimum_test.cc62 node.temporaries = temporaries_array; in TestMaxMinFloat()
121 node.temporaries = temporaries_array; in TestMaxMinQuantized()
178 node.temporaries = temporaries_array; in TestMaxMinQuantizedInt32()
Dunpack_test.cc89 node.temporaries = temporaries_array; in TestUnpackThreeOutputsFloat()
166 node.temporaries = temporaries_array; in TestUnpackOneOutputFloat()
258 node.temporaries = temporaries_array; in TestUnpackThreeOutputsQuantized()
354 node.temporaries = temporaries_array; in TestUnpackThreeOutputsQuantized32()
Dsoftmax_test.cc68 node.temporaries = temporaries_array; in TestSoftmaxFloat()
135 node.temporaries = temporaries_array; in TestSoftmaxQuantized()
201 node.temporaries = temporaries_array; in TestSoftmaxQuantizedSigned()
Dsplit_test.cc86 node.temporaries = temporaries_array; in TestSplitTwoOutputsFloat()
189 node.temporaries = temporaries_array; in TestSplitFourOutputsFloat()
287 node.temporaries = temporaries_array; in TestSplitTwoOutputsQuantized()
376 node.temporaries = temporaries_array; in TestSplitTwoOutputsQuantized32()
Delementwise_test.cc69 node.temporaries = temporaries_array; in TestElementwiseFloat()
131 node.temporaries = temporaries_array; in TestElementwiseBool()
Dconcatenation_test.cc67 node.temporaries = temporaries_array; in TestConcatenateTwoInputs()
129 node.temporaries = temporaries_array; in TestConcatenateQuantizedTwoInputs()
/external/libvpx/libvpx/vpx_dsp/arm/
Didct_neon.asm17 ; non-high-bitdepth builds. q0-q3 are used as temporaries in high-bitdepth.
32 ; $dst0..3 are d registers. q0-q3 are used as temporaries in high-bitdepth.
/external/libvpx/config/arm-neon/vpx_dsp/arm/
Didct_neon.asm.S20 @ non-high-bitdepth builds. q0-q3 are used as temporaries in high-bitdepth.
34 @ $dst0..3 are d registers. q0-q3 are used as temporaries in high-bitdepth.

12345