Home
last modified time | relevance | path

Searched refs:TfLiteIntArrayCreate (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/lite/
Dcontext_test.cc28 TfLiteIntArray* a = TfLiteIntArrayCreate(0); in TEST()
29 TfLiteIntArray* b = TfLiteIntArrayCreate(3); in TEST()
35 TfLiteIntArray* a = TfLiteIntArrayCreate(2); 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()
Dmodel_test.cc165 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()
Dcontext.c20 TfLiteIntArray* TfLiteIntArrayCreate(int size) { in TfLiteIntArrayCreate() function
49 TfLiteIntArray* ret = TfLiteIntArrayCreate(src->size); in TfLiteIntArrayCopy()
Dinterpreter.cc164 plan_cache_.reset(TfLiteIntArrayCreate(execution_plan_.size())); in GetExecutionPlan()
241 TfLiteIntArray* lite = TfLiteIntArrayCreate(x.size()); in convertVectorToTfLiteIntArray()
293 node.temporaries = TfLiteIntArrayCreate(0); in AddNodeWithParameters()
Dstring_util.cc96 auto dims = TfLiteIntArrayCreate(1); in WriteToTensor()
Dcontext.h62 TfLiteIntArray* TfLiteIntArrayCreate(int size);
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dmean.cc62 TfLiteIntArray* axis_size = TfLiteIntArrayCreate(1); in ResizeTempAxis()
75 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_num_dims); in ResizeOutputTensor()
113 TfLiteIntArrayCreate(input_num_dims - num_reduce_axis); in ResizeOutputTensor()
138 node->temporaries = TfLiteIntArrayCreate(2); in InitializeTemporaries()
143 TfLiteIntArray* index_size = TfLiteIntArrayCreate(1); in InitializeTemporaries()
Dsvdf.cc90 TfLiteIntArray* state_size_array = TfLiteIntArrayCreate(2); in Prepare()
100 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); in Prepare()
108 node->temporaries = TfLiteIntArrayCreate(1); in Prepare()
111 TfLiteIntArray* scratch_size_array = TfLiteIntArrayCreate(2); in Prepare()
Dbidirectional_sequence_rnn.cc86 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()
Dhashtable_lookup.cc81 TfLiteIntArray* hitSize = TfLiteIntArrayCreate(1); in Prepare()
89 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value)); in Prepare()
Dbasic_rnn.cc67 TfLiteIntArray* hidden_state_size_array = TfLiteIntArrayCreate(2); in Prepare()
77 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); in Prepare()
Dlstm.cc261 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()
Dunidirectional_sequence_lstm.cc262 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()
Dunidirectional_sequence_rnn.cc72 TfLiteIntArray* hidden_state_size_array = TfLiteIntArrayCreate(2); in Prepare()
82 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(3); in Prepare()
Dconv.cc193 TfLiteIntArray* output_size = TfLiteIntArrayCreate(4); in Prepare()
230 node->temporaries = TfLiteIntArrayCreate(temporaries_count); in Prepare()
235 TfLiteIntArray* im2col_size = TfLiteIntArrayCreate(4); in Prepare()
253 TfLiteIntArray* hwcn_weights_size = TfLiteIntArrayCreate(2); in Prepare()
Dtopk_v2.cc50 TfLiteIntArray* output_indexes_shape = TfLiteIntArrayCreate(num_dimensions); in ResizeOutput()
51 TfLiteIntArray* output_values_shape = TfLiteIntArrayCreate(num_dimensions); in ResizeOutput()
Dreshape.cc50 TfLiteIntArray* output_size = TfLiteIntArrayCreate(params->num_dimensions); in Prepare()
Dembedding_lookup.cc62 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value)); in Prepare()
Dlocal_response_norm.cc49 TfLiteIntArray* output_size = TfLiteIntArrayCreate(4); in Prepare()
Dsqueeze.cc72 TfLiteIntArrayCreate(input_num_dims - num_squeezed_dims); in Prepare()
Dspace_to_depth.cc65 TfLiteIntArray* output_size = TfLiteIntArrayCreate(4); in Prepare()
Dresize_bilinear.cc41 TfLiteIntArray* output_size = TfLiteIntArrayCreate(4); in ResizeOutputTensor()
Dkernel_util.cc102 TfLiteIntArrayCreate(out_dims), TfLiteIntArrayFree); in CalculateShapeForBroadcast()
/external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/
Dextract_feature.cc62 TfLiteIntArray* outputSize1 = TfLiteIntArrayCreate(1); in Prepare()
63 TfLiteIntArray* outputSize2 = TfLiteIntArrayCreate(1); in Prepare()
Dpredict.cc104 TfLiteIntArray* label_size = TfLiteIntArrayCreate(1); in Prepare()
106 TfLiteIntArray* weight_size = TfLiteIntArrayCreate(1); in Prepare()

12