Home
last modified time | relevance | path

Searched refs:TfLiteTensor (Results 1 – 25 of 468) sorted by relevance

12345678910>>...19

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h99 const TfLiteTensor* input, const TfLiteTensor* input_to_input_weights,
100 const TfLiteTensor* input_to_forget_weights,
101 const TfLiteTensor* input_to_cell_weights,
102 const TfLiteTensor* input_to_output_weights,
103 const TfLiteTensor* recurrent_to_input_weights,
104 const TfLiteTensor* recurrent_to_forget_weights,
105 const TfLiteTensor* recurrent_to_cell_weights,
106 const TfLiteTensor* recurrent_to_output_weights,
107 const TfLiteTensor* cell_to_input_weights,
108 const TfLiteTensor* cell_to_forget_weights,
[all …]
Dkernel_util.h40 const TfLiteTensor* GetInput(const TfLiteContext* context,
52 int index, const TfLiteTensor** tensor);
61 TfLiteTensor* GetVariableInput(TfLiteContext* context, const TfLiteNode* node,
71 TfLiteTensor* GetOutput(TfLiteContext* context, const TfLiteNode* node,
83 int index, TfLiteTensor** tensor);
94 const TfLiteTensor* GetOptionalInputTensor(const TfLiteContext* context,
105 TfLiteTensor* GetTemporary(TfLiteContext* context, const TfLiteNode* node,
120 TfLiteTensor** tensor);
129 const TfLiteTensor* GetIntermediates(TfLiteContext* context,
144 TfLiteTensor** tensor);
[all …]
Dlstm_eval_test.cc57 TfLiteTensor* Geti2i() { in Geti2i()
62 TfLiteTensor* Geti2f() { in Geti2f()
67 TfLiteTensor* Geti2c() { in Geti2c()
72 TfLiteTensor* Geti2o() { in Geti2o()
77 TfLiteTensor* Getr2i() { in Getr2i()
82 TfLiteTensor* Getr2f() { in Getr2f()
87 TfLiteTensor* Getr2c() { in Getr2c()
92 TfLiteTensor* Getr2o() { in Getr2o()
97 TfLiteTensor* GetProjection() { in GetProjection()
129 void PackWeightToTensor(TfLiteTensor* tensor, std::vector<T>& data, in PackWeightToTensor()
[all …]
Dbasic_rnn.cc63 const TfLiteTensor* input; in Prepare()
65 const TfLiteTensor* input_weights; in Prepare()
68 const TfLiteTensor* recurrent_weights; in Prepare()
72 const TfLiteTensor* bias; in Prepare()
74 const TfLiteTensor* hidden_state; in Prepare()
96 TfLiteTensor* output; in Prepare()
117 TfLiteTensor* input_quantized; in Prepare()
128 TfLiteTensor* hidden_state_quantized; in Prepare()
142 TfLiteTensor* scaling_factors; in Prepare()
155 TfLiteTensor* accum_scratch; in Prepare()
[all …]
Dbidirectional_sequence_lstm.cc196 const TfLiteTensor* input_to_forget_weights; in CheckLstmTensorDimensionsAndTypes()
207 const TfLiteTensor* input_to_input_weights = in CheckLstmTensorDimensionsAndTypes()
217 const TfLiteTensor* input_to_cell_weights; in CheckLstmTensorDimensionsAndTypes()
227 const TfLiteTensor* input_to_output_weights; in CheckLstmTensorDimensionsAndTypes()
237 const TfLiteTensor* recurrent_to_input_weights = in CheckLstmTensorDimensionsAndTypes()
249 const TfLiteTensor* recurrent_to_forget_weights; in CheckLstmTensorDimensionsAndTypes()
261 const TfLiteTensor* recurrent_to_cell_weights; in CheckLstmTensorDimensionsAndTypes()
281 const TfLiteTensor* cell_to_input_weights = in CheckLstmTensorDimensionsAndTypes()
290 const TfLiteTensor* cell_to_forget_weights = in CheckLstmTensorDimensionsAndTypes()
299 const TfLiteTensor* cell_to_output_weights = in CheckLstmTensorDimensionsAndTypes()
[all …]
Dbidirectional_sequence_rnn.cc100 const TfLiteTensor* input; in Prepare()
102 const TfLiteTensor* fw_input_weights; in Prepare()
105 const TfLiteTensor* fw_recurrent_weights; in Prepare()
109 const TfLiteTensor* fw_bias; in Prepare()
112 const TfLiteTensor* fw_hidden_state; in Prepare()
115 const TfLiteTensor* bw_input_weights; in Prepare()
118 const TfLiteTensor* bw_recurrent_weights; in Prepare()
122 const TfLiteTensor* bw_bias; in Prepare()
125 const TfLiteTensor* bw_hidden_state; in Prepare()
129 const TfLiteTensor* aux_input = in Prepare()
[all …]
Dfully_connected.cc57 TfLiteContext* context, TfLiteTensor* ledger) { in CreateLedgerTensor()
129 const TfLiteTensor* input, in CheckTypes()
130 const TfLiteTensor* filter, in CheckTypes()
131 const TfLiteTensor* bias, TfLiteTensor* output, in CheckTypes()
201 const TfLiteTensor* input; in PrepareImpl()
203 const TfLiteTensor* filter; in PrepareImpl()
206 const TfLiteTensor* bias = in PrepareImpl()
210 TfLiteTensor* output; in PrepareImpl()
272 TfLiteTensor* input_quantized; in PrepareImpl()
283 TfLiteTensor* scaling_factors; in PrepareImpl()
[all …]
Dlstm.cc96 TfLiteTensor* ledger) { in make_ledger()
108 TfLiteStatus copy_ledger(const TfLiteSparsity* sparsity, TfLiteTensor* ledger) { in copy_ledger()
149 const TfLiteTensor* cell_state = in PopulateQuantizedLstmParams8x8_16()
152 TfLiteTensor* output_tensor; in PopulateQuantizedLstmParams8x8_16()
178 const TfLiteTensor* input; in PopulateQuantizedLstmParams8x8_16()
181 const TfLiteTensor* input_to_input_weights = in PopulateQuantizedLstmParams8x8_16()
183 const TfLiteTensor* input_to_forget_weights; in PopulateQuantizedLstmParams8x8_16()
187 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_16()
191 const TfLiteTensor* input_to_output_weights; in PopulateQuantizedLstmParams8x8_16()
196 const TfLiteTensor* recurrent_to_input_weights = in PopulateQuantizedLstmParams8x8_16()
[all …]
Dunidirectional_sequence_rnn.cc64 const TfLiteTensor* input; in Prepare()
66 const TfLiteTensor* input_weights; in Prepare()
69 const TfLiteTensor* recurrent_weights; in Prepare()
73 const TfLiteTensor* bias; in Prepare()
75 const TfLiteTensor* hidden_state; in Prepare()
102 TfLiteTensor* output; in Prepare()
124 TfLiteTensor* input_quantized; in Prepare()
135 TfLiteTensor* hidden_state_quantized; in Prepare()
149 TfLiteTensor* scaling_factors; in Prepare()
162 TfLiteTensor* accum_scratch; in Prepare()
[all …]
Dunidirectional_sequence_lstm.cc57 const TfLiteTensor* cell_state = in PopulateQuantizedLstmParams8x8_16()
60 TfLiteTensor* output_tensor; in PopulateQuantizedLstmParams8x8_16()
87 const TfLiteTensor* input; in PopulateQuantizedLstmParams8x8_16()
91 const TfLiteTensor* input_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
93 const TfLiteTensor* input_to_forget_weights; in PopulateQuantizedLstmParams8x8_16()
98 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_16()
102 const TfLiteTensor* input_to_output_weights; in PopulateQuantizedLstmParams8x8_16()
108 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
110 const TfLiteTensor* recurrent_to_forget_weights; in PopulateQuantizedLstmParams8x8_16()
115 const TfLiteTensor* recurrent_to_cell_weights; in PopulateQuantizedLstmParams8x8_16()
[all …]
Dlstm_eval.cc113 inline float GetTensorScale(const TfLiteTensor* tensor) { in GetTensorScale()
1688 const TfLiteTensor* input, const TfLiteTensor* input_to_input_weights, in EvalFloat()
1689 const TfLiteTensor* input_to_forget_weights, in EvalFloat()
1690 const TfLiteTensor* input_to_cell_weights, in EvalFloat()
1691 const TfLiteTensor* input_to_output_weights, in EvalFloat()
1692 const TfLiteTensor* recurrent_to_input_weights, in EvalFloat()
1693 const TfLiteTensor* recurrent_to_forget_weights, in EvalFloat()
1694 const TfLiteTensor* recurrent_to_cell_weights, in EvalFloat()
1695 const TfLiteTensor* recurrent_to_output_weights, in EvalFloat()
1696 const TfLiteTensor* cell_to_input_weights, in EvalFloat()
[all …]
Dkernel_util.cc36 inline TfLiteTensor* GetTensorAtIndex(const TfLiteContext* context, in GetTensorAtIndex()
80 inline TfLiteTensor* GetMutableInput(const TfLiteContext* context, in GetMutableInput()
92 const TfLiteTensor** tensor) { in GetMutableInputSafe()
103 const TfLiteTensor* GetInput(const TfLiteContext* context, in GetInput()
109 int index, const TfLiteTensor** tensor) { in GetInputSafe()
113 TfLiteTensor* GetVariableInput(TfLiteContext* context, const TfLiteNode* node, in GetVariableInput()
115 TfLiteTensor* tensor = GetMutableInput(context, node, index); in GetVariableInput()
119 TfLiteTensor* GetOutput(TfLiteContext* context, const TfLiteNode* node, in GetOutput()
130 int index, TfLiteTensor** tensor) { in GetOutputSafe()
139 const TfLiteTensor* GetOptionalInputTensor(const TfLiteContext* context, in GetOptionalInputTensor()
[all …]
Dvariable_ops_test.cc78 TfLiteTensor* input_assign_index = interpreter_.tensor(0); in TEST_F()
80 TfLiteTensor* input_read_index = interpreter_.tensor(1); in TEST_F()
82 TfLiteTensor* input_data_index = interpreter_.tensor(2); in TEST_F()
87 TfLiteTensor* output = interpreter_.tensor(3); in TEST_F()
94 TfLiteTensor* input_assign_index = interpreter_.tensor(0); in TEST_F()
96 TfLiteTensor* input_read_index = interpreter_.tensor(1); in TEST_F()
98 TfLiteTensor* input_data_index = interpreter_.tensor(2); in TEST_F()
110 TfLiteTensor* input_assign_index = interpreter_.tensor(0); in TEST_F()
112 TfLiteTensor* input_read_index = interpreter_.tensor(1); in TEST_F()
114 TfLiteTensor* input_data_index = interpreter_.tensor(2); in TEST_F()
[all …]
Dbatch_matmul.cc78 const TfLiteTensor* lhs;
79 const TfLiteTensor* rhs;
80 TfLiteTensor* output;
103 TfLiteTensor* output) { in ResizeOutputTensor()
130 const TfLiteTensor* lhs = op_context->lhs; in InitializeTemporaries()
131 const TfLiteTensor* rhs = op_context->rhs; in InitializeTemporaries()
155 TfLiteTensor* scratch_buffer; in InitializeTemporaries()
178 TfLiteTensor* scratch_buffer; in InitializeTemporaries()
181 const TfLiteTensor* rhs = op_context->rhs; in InitializeTemporaries()
219 TfLiteTensor* input_quantized; in InitializeTemporaries()
[all …]
Dtranspose_conv.cc104 const TfLiteTensor* shape_tensor, in ResizeTensor()
105 TfLiteTensor* tensor_to_resize) { in ResizeTensor()
168 const TfLiteTensor* output_shape, in ResizeCol2ImTensor()
169 const TfLiteTensor* weights, in ResizeCol2ImTensor()
170 const TfLiteTensor* input, in ResizeCol2ImTensor()
171 TfLiteTensor* col2im) { in ResizeCol2ImTensor()
191 const TfLiteTensor* weights, in ResizeAndTransposeWeights()
192 TfLiteTensor* transposed_weights) { in ResizeAndTransposeWeights()
251 const TfLiteTensor* output_shape; in Prepare()
254 const TfLiteTensor* weights; in Prepare()
[all …]
Dlsh_projection.cc76 const TfLiteTensor* hash; in Resize()
82 const TfLiteTensor* input; in Resize()
87 const TfLiteTensor* weight; in Resize()
94 TfLiteTensor* output; in Resize()
115 int RunningSignBit(const TfLiteTensor* input, const TfLiteTensor* weight, in RunningSignBit()
145 void SparseLshProjection(const TfLiteTensor* hash, const TfLiteTensor* input, in SparseLshProjection()
146 const TfLiteTensor* weight, int32_t* out_buf) { in SparseLshProjection()
160 void DenseLshProjection(const TfLiteTensor* hash, const TfLiteTensor* input, in DenseLshProjection()
161 const TfLiteTensor* weight, int32_t* out_buf) { in DenseLshProjection()
177 TfLiteTensor* out_tensor; in Eval()
[all …]
Dscatter_nd.cc37 const TfLiteTensor* shape, in ResizeOutputTensor()
38 TfLiteTensor* output) { in ResizeOutputTensor()
77 const TfLiteTensor* indices; in Prepare()
79 const TfLiteTensor* updates; in Prepare()
81 const TfLiteTensor* shape; in Prepare()
102 TfLiteTensor* output; in Prepare()
129 TfLiteStatus ScatterNd(const TfLiteTensor* indices, const TfLiteTensor* updates, in ScatterNd()
130 TfLiteTensor* output) { in ScatterNd()
139 TfLiteStatus EvalScatterNd(TfLiteContext* context, const TfLiteTensor* indices, in EvalScatterNd()
140 const TfLiteTensor* updates, in EvalScatterNd()
[all …]
Dgather_nd.cc36 const TfLiteTensor* params; in Prepare()
38 const TfLiteTensor* indices; in Prepare()
40 TfLiteTensor* output; in Prepare()
105 TfLiteStatus GatherNd(const TfLiteTensor* params, const TfLiteTensor* indices, in GatherNd()
106 TfLiteTensor* output) { in GatherNd()
115 TfLiteStatus GatherNdString(const TfLiteTensor* params, in GatherNdString()
116 const TfLiteTensor* indices, TfLiteTensor* output) { in GatherNdString()
124 TfLiteStatus EvalGatherNd(TfLiteContext* context, const TfLiteTensor* params, in EvalGatherNd()
125 const TfLiteTensor* indices, TfLiteTensor* output) { in EvalGatherNd()
150 const TfLiteTensor* params; in Eval()
[all …]
Dsparse_to_dense.cc40 TfLiteStatus Resize(TfLiteContext* context, const TfLiteTensor* output_shape, in Resize()
41 TfLiteTensor* output) { in Resize()
52 const TfLiteTensor* indices, in CheckDimensionsMatch()
53 const TfLiteTensor* output_shape, in CheckDimensionsMatch()
54 const TfLiteTensor* values) { in CheckDimensionsMatch()
86 const TfLiteTensor* indices, in GetIndicesVector()
129 const TfLiteTensor* output_shape, in ResizeOutputShape()
130 TfLiteTensor* output) { in ResizeOutputShape()
146 const TfLiteTensor* indices; in Prepare()
149 const TfLiteTensor* output_shape; in Prepare()
[all …]
Dnon_max_suppression.cc62 TfLiteStatus SetTensorSizes(TfLiteContext* context, TfLiteTensor* tensor, in SetTensorSizes()
82 const TfLiteTensor* input_boxes; in Prepare()
89 const TfLiteTensor* input_scores; in Prepare()
97 const TfLiteTensor* input_max_output_size; in Prepare()
111 const TfLiteTensor* input_iou_threshold; in Prepare()
117 const TfLiteTensor* input_score_threshold; in Prepare()
125 const TfLiteTensor* input_sigma; in Prepare()
132 TfLiteTensor* output_selected_indices; in Prepare()
138 TfLiteTensor* output_selected_scores; in Prepare()
143 TfLiteTensor* output_num_selected_indices; in Prepare()
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc332 const TfLiteTensor* input, const TfLiteTensor* input_to_input_weights, in EvalFloat()
333 const TfLiteTensor* input_to_forget_weights, in EvalFloat()
334 const TfLiteTensor* input_to_cell_weights, in EvalFloat()
335 const TfLiteTensor* input_to_output_weights, in EvalFloat()
336 const TfLiteTensor* recurrent_to_input_weights, in EvalFloat()
337 const TfLiteTensor* recurrent_to_forget_weights, in EvalFloat()
338 const TfLiteTensor* recurrent_to_cell_weights, in EvalFloat()
339 const TfLiteTensor* recurrent_to_output_weights, in EvalFloat()
340 const TfLiteTensor* cell_to_input_weights, in EvalFloat()
341 const TfLiteTensor* cell_to_forget_weights, in EvalFloat()
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc264 const TfLiteTensor* input, const TfLiteTensor* input_to_input_weights, in EvalCalibration()
265 const TfLiteTensor* input_to_forget_weights, in EvalCalibration()
266 const TfLiteTensor* input_to_cell_weights, in EvalCalibration()
267 const TfLiteTensor* input_to_output_weights, in EvalCalibration()
268 const TfLiteTensor* recurrent_to_input_weights, in EvalCalibration()
269 const TfLiteTensor* recurrent_to_forget_weights, in EvalCalibration()
270 const TfLiteTensor* recurrent_to_cell_weights, in EvalCalibration()
271 const TfLiteTensor* recurrent_to_output_weights, in EvalCalibration()
272 const TfLiteTensor* cell_to_input_weights, in EvalCalibration()
273 const TfLiteTensor* cell_to_forget_weights, in EvalCalibration()
[all …]
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc30 void GruImpl(const TfLiteTensor* input, const TfLiteTensor* input_state, in GruImpl()
31 const TfLiteTensor* gate_weight, const TfLiteTensor* gate_bias, in GruImpl()
32 const TfLiteTensor* candidate_weight, in GruImpl()
33 const TfLiteTensor* candidate_bias, TfLiteTensor* output, in GruImpl()
34 TfLiteTensor* output_state, TfLiteTensor* activation, in GruImpl()
35 TfLiteTensor* concat, in GruImpl()
130 const TfLiteTensor* input; in Prepare()
138 const TfLiteTensor* input_state; in Prepare()
146 const TfLiteTensor* gate_weight; in Prepare()
154 const TfLiteTensor* gate_bias; in Prepare()
[all …]
/external/tensorflow/tensorflow/lite/c/
Dc_api.h91 typedef struct TfLiteTensor TfLiteTensor; typedef
166 TFL_CAPI_EXPORT extern TfLiteTensor* TfLiteInterpreterGetInputTensor(
230 TFL_CAPI_EXPORT extern const TfLiteTensor* TfLiteInterpreterGetOutputTensor(
242 TFL_CAPI_EXPORT extern TfLiteType TfLiteTensorType(const TfLiteTensor* tensor);
245 TFL_CAPI_EXPORT extern int32_t TfLiteTensorNumDims(const TfLiteTensor* tensor);
249 TFL_CAPI_EXPORT extern int32_t TfLiteTensorDim(const TfLiteTensor* tensor,
253 TFL_CAPI_EXPORT extern size_t TfLiteTensorByteSize(const TfLiteTensor* tensor);
261 TFL_CAPI_EXPORT extern void* TfLiteTensorData(const TfLiteTensor* tensor);
264 TFL_CAPI_EXPORT extern const char* TfLiteTensorName(const TfLiteTensor* tensor);
271 const TfLiteTensor* tensor);
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util_test.cc50 void PopulateContext(std::vector<TfLiteTensor>& tensors, in PopulateContext()
85 TfLiteTensor CreateQuantizedTensor(const int8_t* data, TfLiteIntArray* dims, in CreateQuantizedTensor()
88 TfLiteTensor result; in CreateQuantizedTensor()
104 TfLiteTensor CreateQuantizedTensor(const uint8_t* data, TfLiteIntArray* dims, in CreateQuantizedTensor()
107 TfLiteTensor result; in CreateQuantizedTensor()
123 TfLiteTensor CreateTensor(TfLiteIntArray* dims, const char* name, in CreateTensor()
125 TfLiteTensor result; in CreateTensor()
138 TfLiteTensor CreateFloatTensor(const float* data, TfLiteIntArray* dims, in CreateFloatTensor()
140 TfLiteTensor result = CreateTensor(dims, name, is_variable); in CreateFloatTensor()
153 TfLiteTensor input = CreateQuantizedTensor( in TEST()
[all …]

12345678910>>...19