Home
last modified time | relevance | path

Searched refs:kTfLiteInt32 (Results 1 – 25 of 168) sorted by relevance

1234567

/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc106 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildAddSubgraph()
107 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildAddSubgraph()
108 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildAddSubgraph()
137 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildMulSubgraph()
138 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildMulSubgraph()
139 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildMulSubgraph()
168 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildPadSubgraph()
169 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildPadSubgraph()
170 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildPadSubgraph()
200 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildIfSubgraph()
[all …]
Darg_min_max.cc79 axis->type == kTfLiteInt32 || axis->type == kTfLiteInt64); in Prepare()
87 case kTfLiteInt32: in Prepare()
88 output->type = kTfLiteInt32; in Prepare()
104 case kTfLiteInt32: in Prepare()
144 if (axis->type == kTfLiteInt32) { in Eval()
146 case kTfLiteInt32: { in Eval()
157 case kTfLiteInt32: in Eval()
182 case kTfLiteInt32: in Eval()
204 case kTfLiteInt32: { in Eval()
215 case kTfLiteInt32: in Eval()
[all …]
Dcall_once_test.cc54 ASSERT_EQ(output->type, kTfLiteInt32); in TEST_F()
68 ASSERT_EQ(output->type, kTfLiteInt32); in TEST_F()
83 ASSERT_EQ(output->type, kTfLiteInt32); in TEST_F()
95 ASSERT_EQ(output->type, kTfLiteInt32); in TEST_F()
110 ASSERT_EQ(output->type, kTfLiteInt32); in TEST_F()
Dbucketize.cc67 if (input->type != kTfLiteInt32 && input->type != kTfLiteFloat32 && in Prepare()
77 output->type = kTfLiteInt32; in Prepare()
104 TF_LITE_ENSURE_TYPES_EQ(context, output->type, kTfLiteInt32); in BucketizeImpl()
124 case kTfLiteInt32: { in Eval()
Dshape.cc51 case kTfLiteInt32: in Prepare()
52 output->type = kTfLiteInt32; in Prepare()
80 case kTfLiteInt32: in Prepare()
Dsparse_to_dense.cc131 if (output_shape->type == kTfLiteInt32) { in ResizeOutputShape()
173 context, indices->type == kTfLiteInt32 || indices->type == kTfLiteInt64); in Prepare()
174 TF_LITE_ENSURE(context, output_shape->type == kTfLiteInt32 || in Prepare()
176 TF_LITE_ENSURE(context, values->type == kTfLiteInt32 || in Prepare()
241 case kTfLiteInt32: { in EvalForIndexType()
267 case kTfLiteInt32: in Eval()
Dreverse_sequence.cc46 if (input->type != kTfLiteInt32 && input->type != kTfLiteFloat32 && in Prepare()
55 if (seq_lengths->type != kTfLiteInt32 && seq_lengths->type != kTfLiteInt64) { in Prepare()
114 case kTfLiteInt32: { in ReverseSequenceHelper()
146 case kTfLiteInt32: { in Eval()
Dcumsum.cc40 TF_LITE_ENSURE(context, input->type == kTfLiteInt32 || in Prepare()
43 TF_LITE_ENSURE_EQ(context, axis->type, kTfLiteInt32); in Prepare()
72 case kTfLiteInt32: { in Eval()
Dslice.cc94 if (begin->type == kTfLiteInt32) { in ResizeOutputShape()
137 begin->type == kTfLiteInt32 || begin->type == kTfLiteInt64); in Prepare()
139 size->type == kTfLiteInt32 || size->type == kTfLiteInt64); in Prepare()
189 if (begin->type == kTfLiteInt32) { in Eval()
232 case kTfLiteInt32: in Eval()
Drandom_uniform_custom.cc51 if (tensor->type == kTfLiteInt32) { in CreateDimensionsFromTensor()
76 input->type == kTfLiteInt32 || input->type == kTfLiteInt64); in Prepare()
122 case kTfLiteInt32: in IntValueFromTensor()
153 case kTfLiteInt32: in EvalInt()
Dgather_nd.cc51 case kTfLiteInt32: in Prepare()
62 case kTfLiteInt32: in Prepare()
150 case kTfLiteInt32: in EvalGatherNd()
187 case kTfLiteInt32: in Eval()
Dgather.cc52 case kTfLiteInt32: in Prepare()
71 case kTfLiteInt32: in Prepare()
186 if (positions->type == kTfLiteInt32) { in Eval()
200 case kTfLiteInt32: in Eval()
231 case kTfLiteInt32: in Eval()
Dnon_max_suppression.cc101 TF_LITE_ENSURE_EQ(context, input_max_output_size->type, kTfLiteInt32); in Prepare()
137 output_selected_indices->type = kTfLiteInt32; in Prepare()
148 output_num_selected_indices->type = kTfLiteInt32; in Prepare()
164 output_selected_indices->type = kTfLiteInt32; in Prepare()
169 output_num_selected_indices->type = kTfLiteInt32; in Prepare()
Dtile.cc65 case kTfLiteInt32: in ResizeOutput()
183 case kTfLiteInt32: in Tile()
203 case kTfLiteInt32: in TileString()
235 if (multipliers->type != kTfLiteInt32 && multipliers->type != kTfLiteInt64) { in Prepare()
277 case kTfLiteInt32: in Eval()
Droll.cc37 TFLITE_DCHECK(t->type == kTfLiteInt32 || t->type == kTfLiteInt64); in ExtractIntegerVector()
40 if (t->type == kTfLiteInt32) { in ExtractIntegerVector()
167 context, (shift->type == kTfLiteInt32) || (shift->type == kTfLiteInt64)); in Prepare()
169 (axis->type == kTfLiteInt32) || (axis->type == kTfLiteInt64)); in Prepare()
231 case kTfLiteInt32: in Eval()
Dsegment_sum.cc77 data->type == kTfLiteInt32 || data->type == kTfLiteFloat32); in Prepare()
78 TF_LITE_ENSURE_EQ(context, segment_ids->type, kTfLiteInt32); in Prepare()
110 case kTfLiteInt32: in Eval()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc178 {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteInt32}}) && in IsNodeSupportedByHexagon()
183 node, context, {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteInt32}}) || in IsNodeSupportedByHexagon()
198 node, context, {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteInt32}}) && in IsNodeSupportedByHexagon()
205 {kTfLiteInt32, kTfLiteNoType}})) { in IsNodeSupportedByHexagon()
258 {{kTfLiteInt32}, in IsNodeSupportedByHexagon()
264 {{kTfLiteInt32}, in IsNodeSupportedByHexagon()
267 {kTfLiteInt32}})) in IsNodeSupportedByHexagon()
283 {kTfLiteInt32}})) in IsNodeSupportedByHexagon()
297 {kTfLiteInt32}})) in IsNodeSupportedByHexagon()
353 {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteInt32}}) && in IsNodeSupportedByHexagon()
[all …]
/external/tensorflow/tensorflow/lite/delegates/flex/training/
Dtraining_delegate_test.cc65 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
80 ASSERT_EQ(GetType(2), kTfLiteInt32); in TEST_F()
84 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
99 ASSERT_EQ(GetType(2), kTfLiteInt32); in TEST_F()
127 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version_test.cc127 std::vector<TfLiteType>{kTfLiteInt32, kTfLiteInt32, kTfLiteInt32}), in TEST()
238 .inputs = CreateOpSignatureTensorSpecs(kTfLiteInt32), in TEST()
264 .inputs = CreateOpSignatureTensorSpecs(kTfLiteInt32), in TEST()
689 .inputs = CreateOpSignatureTensorSpecs(kTfLiteInt32), in TEST()
711 std::vector<TfLiteType>{kTfLiteInt32, kTfLiteInt8, kTfLiteInt8}), in TEST()
718 kTfLiteInt32, kTfLiteInt8, kTfLiteInt8, kTfLiteInt32}), in TEST()
726 kTfLiteInt32, kTfLiteInt8, kTfLiteInt8, none_type}), in TEST()
759 kTfLiteInt8, kTfLiteInt8, kTfLiteInt32, kTfLiteInt32, kTfLiteInt16}), in TEST()
814 .inputs = CreateOpSignatureTensorSpecs(kTfLiteInt32), in TEST()
855 std::vector<TfLiteType>{kTfLiteInt32, kTfLiteInt32}), in TEST()
[all …]
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_test.cc87 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
102 ASSERT_EQ(GetType(2), kTfLiteInt32); in TEST_F()
360 SetConstTensor(1, {1}, kTfLiteInt32, in TEST_F()
410 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
425 ASSERT_EQ(GetType(2), kTfLiteInt32); in TEST_F()
463 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
/external/tensorflow/tensorflow/lite/kernels/gradient/
Dbcast_grad_args.cc48 input1->type == kTfLiteInt32 || input1->type == kTfLiteInt64); in Prepare()
85 if (input1->type == kTfLiteInt32) { in Invoke()
92 if (input2->type == kTfLiteInt32) { in Invoke()
194 if (output1->type == kTfLiteInt32) { in Invoke()
208 if (output2->type == kTfLiteInt32) { in Invoke()
/external/armnn/delegate/classic/src/
DArgMinMax.hpp53 case kTfLiteInt32: in VisitArgMinMaxOperator()
70 …if (argMaxParameters->output_type != kTfLiteInt32 && argMaxParameters->output_type != kTfLiteInt64) in VisitArgMinMaxOperator()
83 …if (argMinParameters->output_type != kTfLiteInt32 && argMinParameters->output_type != kTfLiteInt64) in VisitArgMinMaxOperator()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2exec/
Dtfl_while_op.mlir12 // CHECK: Tensor 0 pconst kTfLiteInt32 kTfLiteMmapRo 4 / 0.00 (null) [{{.*}})
13 // CHECK-NEXT: Tensor 1 N kTfLiteInt32 kTfLiteMmapRo 4 / 0.00 (null) [{{.*}})
15 // CHECK-NEXT: Tensor 3 tfl.while kTfLiteInt32 kTfLiteArenaRw 4 / 0.00 (null) [{{.*}})
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Dcall_test.cc59 SetupTensor(subgraph, idx, kTfLiteInt32); in BuildCallSubgraph()
62 SetupTensor(subgraph, idx, kTfLiteInt32); in BuildCallSubgraph()
107 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildGraphWithMultipleOutputs()
108 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildGraphWithMultipleOutputs()
109 SetupTensor(subgraph, kMulOutput, kTfLiteInt32); in BuildGraphWithMultipleOutputs()
110 SetupTensor(subgraph, kAddOutput, kTfLiteInt32); in BuildGraphWithMultipleOutputs()
149 SetupTensor(this_subgraph, idx, kTfLiteInt32); in BuildMultiNodeGraph()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc89 case kTfLiteInt32: in GetFirstDimensionSize()
125 case kTfLiteInt32: { in TensorShapeFromTensor()
163 case kTfLiteInt32: in GetMaxWidthValueRowID()
194 case kTfLiteInt32: in GetMaxWidthRowSplit()
345 case kTfLiteInt32: in CalculateOutputIndexValueRowID()
390 case kTfLiteInt32: in CalculateOutputIndexRowSplit()
531 case kTfLiteInt32: in SetOutput()
599 if (input_shape.type != kTfLiteInt32 && input_shape.type != kTfLiteInt64) { in Prepare()

1234567