Home
last modified time | relevance | path

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

12345678

/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc111 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildAddSubgraph()
112 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildAddSubgraph()
113 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildAddSubgraph()
142 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildMulSubgraph()
143 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildMulSubgraph()
144 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildMulSubgraph()
173 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildPadSubgraph()
174 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildPadSubgraph()
175 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildPadSubgraph()
205 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildIfSubgraph()
[all …]
Darg_min_max.cc75 axis->type == kTfLiteInt32 || axis->type == kTfLiteInt64); in Prepare()
83 case kTfLiteInt32: in Prepare()
84 output->type = kTfLiteInt32; in Prepare()
100 case kTfLiteInt32: in Prepare()
149 if (axis->type == kTfLiteInt32) { in Eval()
151 case kTfLiteInt32: { in Eval()
162 case kTfLiteInt32: in Eval()
184 case kTfLiteInt32: in Eval()
203 case kTfLiteInt32: { in Eval()
214 case kTfLiteInt32: in Eval()
[all …]
Dcall_once_test.cc51 ASSERT_EQ(output->type, kTfLiteInt32); in TEST_F()
65 ASSERT_EQ(output->type, kTfLiteInt32); in TEST_F()
80 ASSERT_EQ(output->type, kTfLiteInt32); in TEST_F()
Dshape.cc51 case kTfLiteInt32: in Prepare()
52 output->type = kTfLiteInt32; in Prepare()
80 case kTfLiteInt32: in Prepare()
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()
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()
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()
Dgather.cc52 case kTfLiteInt32: in Prepare()
71 case kTfLiteInt32: in Prepare()
149 if (positions->type == kTfLiteInt32) { in Eval()
159 case kTfLiteInt32: in Eval()
183 case kTfLiteInt32: in Eval()
Dgather_nd.cc50 case kTfLiteInt32: in Prepare()
61 case kTfLiteInt32: in Prepare()
135 case kTfLiteInt32: in EvalGatherNd()
159 case kTfLiteInt32: in Eval()
Dslice.cc93 if (begin->type == kTfLiteInt32) { in ResizeOutputShape()
129 begin->type == kTfLiteInt32 || begin->type == kTfLiteInt64); in Prepare()
131 size->type == kTfLiteInt32 || size->type == kTfLiteInt64); in Prepare()
175 if (begin->type == kTfLiteInt32) { in Eval()
218 case kTfLiteInt32: in Eval()
Dscatter_nd.cc89 case kTfLiteInt32: in Prepare()
109 case kTfLiteInt32: in Prepare()
158 case kTfLiteInt32: in EvalScatterNd()
182 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()
Dtile.cc65 case kTfLiteInt32: in ResizeOutput()
176 case kTfLiteInt32: in Tile()
196 case kTfLiteInt32: in TileString()
228 if (multipliers->type != kTfLiteInt32 && multipliers->type != kTfLiteInt64) { in Prepare()
264 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()
Drank.cc39 output->type = kTfLiteInt32; in Prepare()
54 if (output->type == kTfLiteInt32) { in Prepare()
Dreverse.cc45 if (input->type != kTfLiteInt32 && input->type != kTfLiteFloat32 && in Prepare()
54 if (axis->type != kTfLiteInt32) { in Prepare()
111 case kTfLiteInt32: { in Eval()
Dstrided_slice.cc159 TF_LITE_ENSURE_TYPES_EQ(context, op_context.begin->type, kTfLiteInt32); in Prepare()
160 TF_LITE_ENSURE_TYPES_EQ(context, op_context.end->type, kTfLiteInt32); in Prepare()
161 TF_LITE_ENSURE_TYPES_EQ(context, op_context.strides->type, kTfLiteInt32); in Prepare()
211 case kTfLiteInt32: in Eval()
Dfloor_mod.cc74 if (type != kTfLiteInt32 && type != kTfLiteFloat32 && type != kTfLiteInt64) { in Prepare()
100 if (input2->type == kTfLiteInt32 || input2->type == kTfLiteInt64) { in EvalImpl()
140 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()
252 {{kTfLiteInt32}, in IsNodeSupportedByHexagon()
258 {{kTfLiteInt32}, in IsNodeSupportedByHexagon()
261 {kTfLiteInt32}})) in IsNodeSupportedByHexagon()
277 {kTfLiteInt32}})) in IsNodeSupportedByHexagon()
291 {kTfLiteInt32}})) in IsNodeSupportedByHexagon()
341 {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteInt32}}) && in IsNodeSupportedByHexagon()
[all …]
/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter_test.cc81 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, input->type); in TF_LITE_MICRO_TEST()
92 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, output->type); in TF_LITE_MICRO_TEST()
101 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, output->type); in TF_LITE_MICRO_TEST()
171 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, input1->type); in TF_LITE_MICRO_TEST()
175 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, output1->type); in TF_LITE_MICRO_TEST()
178 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, input2->type); in TF_LITE_MICRO_TEST()
182 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, output2->type); in TF_LITE_MICRO_TEST()
193 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, input3->type); in TF_LITE_MICRO_TEST()
197 TF_LITE_MICRO_EXPECT_EQ(kTfLiteInt32, output3->type); in TF_LITE_MICRO_TEST()
280 case kTfLiteInt32: { in TF_LITE_MICRO_TEST()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dgather.cc52 case kTfLiteInt32: in Prepare()
71 case kTfLiteInt32: in Prepare()
149 if (positions->type == kTfLiteInt32) { in Eval()
159 case kTfLiteInt32: in Eval()
183 case kTfLiteInt32: in Eval()
Dfloor_mod.cc61 if (type != kTfLiteInt32 && type != kTfLiteFloat32 && type != kTfLiteInt64) { in Prepare()
77 if (input2->type == kTfLiteInt32 || input2->type == kTfLiteInt64) { in EvalImpl()
117 case kTfLiteInt32: { in Eval()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2exec/
Dtfl_while_op.mlir12 // CHECK: Tensor 0 pconst kTfLiteInt32 kTfLiteMmapRo 4 bytes
13 // CHECK-NEXT: Tensor 1 N kTfLiteInt32 kTfLiteMmapRo 4 bytes
15 // CHECK-NEXT: Tensor 3 tfl.while kTfLiteInt32 kTfLiteArenaRw 4 bytes
/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()
/external/tensorflow/tensorflow/lite/
Dtype_to_tflitetype_test.cc31 EXPECT_EQ(kTfLiteInt32, in TEST()
32 typeToTfLiteType<TfLiteTypeToType<kTfLiteInt32>::Type>()); in TEST()

12345678