| /external/tensorflow/tensorflow/contrib/lite/ |
| D | context.h | 58 } TfLiteIntArray; typedef
62 TfLiteIntArray* TfLiteIntArrayCreate(int size);
65 int TfLiteIntArrayEqual(TfLiteIntArray* a, TfLiteIntArray* b);
69 TfLiteIntArray* TfLiteIntArrayCopy(TfLiteIntArray* src);
72 void TfLiteIntArrayFree(TfLiteIntArray* v);
177 TfLiteIntArray* dims;
203 void TfLiteTensorReset(TfLiteType type, const char* name, TfLiteIntArray* dims,
216 TfLiteIntArray* inputs;
219 TfLiteIntArray* outputs;
224 TfLiteIntArray* temporaries;
[all …]
|
| D | context.c | 20 TfLiteIntArray* TfLiteIntArrayCreate(int size) { in TfLiteIntArrayCreate()
21 TfLiteIntArray* ret = in TfLiteIntArrayCreate()
22 (TfLiteIntArray*)malloc(sizeof(*ret) + sizeof(ret->data[0]) * size); in TfLiteIntArrayCreate()
27 void TfLiteIntArrayPrint(const char* s, TfLiteIntArray* a) { in TfLiteIntArrayPrint()
37 int TfLiteIntArrayEqual(TfLiteIntArray* a, TfLiteIntArray* b) { in TfLiteIntArrayEqual()
47 TfLiteIntArray* TfLiteIntArrayCopy(TfLiteIntArray* src) { in TfLiteIntArrayCopy()
49 TfLiteIntArray* ret = TfLiteIntArrayCreate(src->size); in TfLiteIntArrayCopy()
56 void TfLiteIntArrayFree(TfLiteIntArray* a) { free(a); } in TfLiteIntArrayFree()
67 void TfLiteTensorReset(TfLiteType type, const char* name, TfLiteIntArray* dims, in TfLiteTensorReset()
|
| D | context_test.cc | 28 TfLiteIntArray* a = TfLiteIntArrayCreate(0); in TEST()
29 TfLiteIntArray* b = TfLiteIntArrayCreate(3); in TEST()
35 TfLiteIntArray* a = TfLiteIntArrayCreate(2); in TEST()
38 TfLiteIntArray* b = TfLiteIntArrayCopy(a); in TEST()
48 TfLiteIntArray* a = TfLiteIntArrayCreate(1); in TEST()
50 TfLiteIntArray* b = TfLiteIntArrayCreate(2); in TEST()
53 TfLiteIntArray* c = TfLiteIntArrayCreate(2); in TEST()
56 TfLiteIntArray* d = TfLiteIntArrayCreate(2); in TEST()
|
| D | interpreter.h | 84 void operator()(TfLiteIntArray* a) { in operator()
323 TfLiteStatus ResizeTensorImpl(TfLiteTensor* tensor, TfLiteIntArray* new_size);
331 TfLiteIntArray* new_size);
343 const TfLiteIntArray* nodes_to_replace);
350 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace);
366 TfLiteStatus GetExecutionPlan(TfLiteIntArray** execution_plan);
371 TfLiteIntArray** execution_plan);
422 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> plan_cache_;
|
| D | interpreter.cc | 111 const TfLiteIntArray* nodes_to_replace) { in ReplaceSubgraphsWithDelegateKernels()
117 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace) { in ReplaceSubgraphsWithDelegateKernels()
162 TfLiteStatus Interpreter::GetExecutionPlan(TfLiteIntArray** execution_plan) { in GetExecutionPlan()
176 TfLiteIntArray** execution_plan) { in GetExecutionPlan()
240 TfLiteIntArray* convertVectorToTfLiteIntArray(const std::vector<int>& x) { in convertVectorToTfLiteIntArray()
241 TfLiteIntArray* lite = TfLiteIntArrayCreate(x.size()); in convertVectorToTfLiteIntArray()
315 TfLiteIntArray* dims_lite = convertVectorToTfLiteIntArray(dims); in ResizeInputTensor()
321 const TfLiteIntArray* tensors) { in HasDynamicTensor()
425 TfLiteIntArray* new_size) { in ResizeTensor()
544 TfLiteIntArray* new_size) { in ResizeTensorImpl()
|
| D | arena_planner.cc | 84 TfLiteIntArray* node_inputs = node.inputs; in PlanAllocations()
105 TfLiteIntArray* node_outputs = node.outputs; in PlanAllocations()
113 TfLiteIntArray* node_inputs = node.inputs; in PlanAllocations()
229 TfLiteIntArray* node_temporaries = node.temporaries; in CalculateAllocationOfInternalTensors()
242 TfLiteIntArray* node_temporaries = node.temporaries; in CalculateDeallocationOfInternalTensors()
|
| D | graph_info.cc | 30 explicit TfLiteIntArrayView(const TfLiteIntArray* int_array) in TfLiteIntArrayView()
41 const TfLiteIntArray* int_array_;
54 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition, in PartitionGraphIntoIndependentSubgraphsImpl()
216 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition, in PartitionGraphIntoIndependentSubgraphs()
|
| D | model_test.cc | 165 TfLiteIntArray* desired_inputs = TfLiteIntArrayCreate(2); in TEST()
168 TfLiteIntArray* desired_outputs = TfLiteIntArrayCreate(1); in TEST()
184 TfLiteIntArray* desired_inputs = TfLiteIntArrayCreate(1); in TEST()
185 TfLiteIntArray* desired_outputs = TfLiteIntArrayCreate(1); in TEST()
|
| D | graph_info_test.cc | 26 TfLiteIntArray* ConvertVector(const std::vector<int>& x) { in ConvertVector()
27 TfLiteIntArray* lite = TfLiteIntArrayCreate(x.size()); in ConvertVector()
78 TfLiteIntArray* nodes_to_partition_int_array = in PartitionGraph()
|
| D | interpreter_test.cc | 364 TfLiteIntArray* newSize = TfLiteIntArrayCopy(tensor0->dims); in TEST()
455 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(1); in TEST()
512 TfLiteIntArray* newSize = TfLiteIntArrayCopy(tensor0->dims); in TEST()
596 TfLiteIntArray* newSize = TfLiteIntArrayCopy(tensor0->dims); in CopyOpRegistration()
704 TfLiteIntArray* newSize = TfLiteIntArrayCopy(tensor0->dims); in AddOpRegistration()
705 TfLiteIntArray* newSizeOther = TfLiteIntArrayCopy(tensor1->dims); in AddOpRegistration()
756 TfLiteIntArray* nodes_to_separate = in SimpleDelegate()
770 TfLiteIntArray* execution_plan; in SimpleDelegate()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | kernel_util_test.cc | 51 std::vector<int> GetShape(TfLiteIntArray* dims) { in GetShape()
88 TfLiteIntArray* output = nullptr; in TEST_F()
97 TfLiteIntArray* output = nullptr; in TEST_F()
112 TfLiteIntArray* output = nullptr; in TEST_F()
129 TfLiteIntArray* output = nullptr; in TEST_F()
|
| D | mean.cc | 62 TfLiteIntArray* axis_size = TfLiteIntArrayCreate(1); in ResizeTempAxis()
71 const TfLiteIntArray* input_dims = op_context->input->dims; in ResizeOutputTensor()
75 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_num_dims); in ResizeOutputTensor()
112 TfLiteIntArray* output_dims = in ResizeOutputTensor()
143 TfLiteIntArray* index_size = TfLiteIntArrayCreate(1); in InitializeTemporaries()
|
| D | kernel_util.cc | 97 TfLiteIntArray** output_shape) { in CalculateShapeForBroadcast()
101 std::unique_ptr<TfLiteIntArray, void (*)(TfLiteIntArray*)> shape( in CalculateShapeForBroadcast()
|
| D | bidirectional_sequence_rnn.cc | 86 TfLiteIntArray* fw_hidden_state_size_array = TfLiteIntArrayCreate(2); in Prepare()
94 TfLiteIntArray* bw_hidden_state_size_array = TfLiteIntArrayCreate(2); in Prepare()
107 TfLiteIntArray* fw_output_size_array = TfLiteIntArrayCreate(3); in Prepare()
113 TfLiteIntArray* bw_output_size_array = TfLiteIntArrayCreate(3); in Prepare()
|
| D | topk_v2.cc | 50 TfLiteIntArray* output_indexes_shape = TfLiteIntArrayCreate(num_dimensions); in ResizeOutput()
51 TfLiteIntArray* output_values_shape = TfLiteIntArrayCreate(num_dimensions); in ResizeOutput()
60 auto resize_tensor = [context](TfLiteTensor* tensor, TfLiteIntArray* new_size, in ResizeOutput()
61 TfLiteIntArray* delete_on_error) { in ResizeOutput()
|
| D | squeeze.cc | 48 const TfLiteIntArray* input_dims = op_context.input->dims; in Prepare()
71 TfLiteIntArray* output_dims = in Prepare()
|
| D | svdf.cc | 90 TfLiteIntArray* state_size_array = TfLiteIntArrayCreate(2); in Prepare()
100 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); in Prepare()
111 TfLiteIntArray* scratch_size_array = TfLiteIntArrayCreate(2); in Prepare()
|
| D | transpose.cc | 59 TfLiteIntArray* input_size = op_context->input->dims; in ResizeOutputTensor()
60 TfLiteIntArray* output_size = TfLiteIntArrayCopy(input_size); in ResizeOutputTensor()
|
| D | unidirectional_sequence_lstm.cc | 262 TfLiteIntArray* output_size = TfLiteIntArrayCreate(3); in Prepare()
269 TfLiteIntArray* output_state_size = TfLiteIntArrayCreate(2); in Prepare()
276 TfLiteIntArray* cell_size = TfLiteIntArrayCreate(2); in Prepare()
290 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
297 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
|
| D | lstm.cc | 261 TfLiteIntArray* output_size = TfLiteIntArrayCreate(2); in Prepare()
267 TfLiteIntArray* output_state_size = TfLiteIntArrayCreate(2); in Prepare()
274 TfLiteIntArray* cell_size = TfLiteIntArrayCreate(2); in Prepare()
288 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
295 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
|
| D | basic_rnn.cc | 67 TfLiteIntArray* hidden_state_size_array = TfLiteIntArrayCreate(2); in Prepare()
77 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); in Prepare()
|
| D | hashtable_lookup.cc | 81 TfLiteIntArray* hitSize = TfLiteIntArrayCreate(1); in Prepare()
89 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value)); in Prepare()
|
| D | batch_to_space_nd.cc | 58 TfLiteIntArray* input_size = op_context->input->dims; in ResizeOutputTensor()
86 TfLiteIntArray* output_size = TfLiteIntArrayCopy(input_size); in ResizeOutputTensor()
|
| D | pad.cc | 60 TfLiteIntArray* input_size = op_context->input->dims; in ResizeOutputTensor()
61 TfLiteIntArray* output_size = TfLiteIntArrayCopy(input_size); in ResizeOutputTensor()
|
| /external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/ |
| D | extract_feature.cc | 62 TfLiteIntArray* outputSize1 = TfLiteIntArrayCreate(1); in Prepare()
63 TfLiteIntArray* outputSize2 = TfLiteIntArrayCreate(1); in Prepare()
|