/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_eval.h | 99 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 …]
|
D | kernel_util.h | 40 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 …]
|
D | lstm_eval_test.cc | 57 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 …]
|
D | basic_rnn.cc | 63 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 …]
|
D | bidirectional_sequence_lstm.cc | 196 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 …]
|
D | bidirectional_sequence_rnn.cc | 100 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 …]
|
D | fully_connected.cc | 57 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 …]
|
D | lstm.cc | 96 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 …]
|
D | unidirectional_sequence_rnn.cc | 64 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 …]
|
D | unidirectional_sequence_lstm.cc | 57 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 …]
|
D | lstm_eval.cc | 113 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 …]
|
D | kernel_util.cc | 36 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 …]
|
D | variable_ops_test.cc | 78 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 …]
|
D | batch_matmul.cc | 78 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 …]
|
D | transpose_conv.cc | 104 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 …]
|
D | lsh_projection.cc | 76 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 …]
|
D | scatter_nd.cc | 37 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 …]
|
D | gather_nd.cc | 36 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 …]
|
D | sparse_to_dense.cc | 40 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 …]
|
D | non_max_suppression.cc | 62 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/ |
D | lstm.cc | 332 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/ |
D | lstm.cc | 264 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/ |
D | unidirectional_sequence_gru.cc | 30 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/ |
D | c_api.h | 91 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/ |
D | quantization_util_test.cc | 50 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 …]
|