| /external/tensorflow/tensorflow/lite/kernels/ |
| D | arg_min_max.cc | 61 axis->type == kTfLiteInt32 || axis->type == kTfLiteInt64); in Prepare()
70 case kTfLiteInt64: in Prepare()
71 output->type = kTfLiteInt64; in Prepare()
149 case kTfLiteInt64: { in Eval()
187 case kTfLiteInt64: { in Eval()
|
| D | gather.cc | 43 case kTfLiteInt64: in Prepare()
61 case kTfLiteInt64: in Prepare()
143 case kTfLiteInt64: in Eval()
153 if (positions->type == kTfLiteInt64) { in Eval()
163 case kTfLiteInt64: in Eval()
|
| D | gather_nd.cc | 42 case kTfLiteInt64: in Prepare()
52 case kTfLiteInt64: in Prepare()
118 case kTfLiteInt64: in EvalGatherNd()
136 case kTfLiteInt64: in Eval()
|
| D | shape.cc | 48 case kTfLiteInt64: in Prepare()
49 output->type = kTfLiteInt64; in Prepare()
74 case kTfLiteInt64: in Eval()
|
| D | reverse_sequence.cc | 42 input->type != kTfLiteInt64) { in Prepare()
49 if (seq_lengths->type != kTfLiteInt32 && seq_lengths->type != kTfLiteInt64) { in Prepare()
104 case kTfLiteInt64: { in ReverseSequenceHelper()
134 case kTfLiteInt64: { in Eval()
|
| D | slice.cc | 89 } else if (begin->type == kTfLiteInt64) { in ResizeOutputShape()
117 begin->type == kTfLiteInt32 || begin->type == kTfLiteInt64); in Prepare()
119 size->type == kTfLiteInt32 || size->type == kTfLiteInt64); in Prepare()
155 } else if (begin->type == kTfLiteInt64) { in Eval()
205 case kTfLiteInt64: in Eval()
|
| D | sparse_to_dense.cc | 137 } else if (output_shape->type == kTfLiteInt64) { in ResizeOutputShape()
171 context, indices->type == kTfLiteInt32 || indices->type == kTfLiteInt64); in Prepare()
173 output_shape->type == kTfLiteInt64); in Prepare()
230 case kTfLiteInt64: { in Eval()
246 case kTfLiteInt64: { in Eval()
|
| D | one_hot.cc | 97 if (op_context.indices->type == kTfLiteInt64) { in OneHotCompute()
130 case kTfLiteInt64: in Prepare()
141 op_context.indices->type == kTfLiteInt64); in Prepare()
172 case kTfLiteInt64: in Eval()
|
| D | tile.cc | 60 case kTfLiteInt64: in ResizeOutput()
126 case kTfLiteInt64: in Tile()
148 if (multipliers->type != kTfLiteInt32 && multipliers->type != kTfLiteInt64) { in Prepare()
182 case kTfLiteInt64: in Eval()
|
| D | fill.cc | 54 case kTfLiteInt64: in ResizeOutput()
80 TF_LITE_ENSURE(context, dtype == kTfLiteInt32 || dtype == kTfLiteInt64); in Prepare()
113 case kTfLiteInt64: in Eval()
|
| D | comparisons.cc | 141 case kTfLiteInt64: in EqualEval()
177 case kTfLiteInt64: in NotEqualEval()
209 case kTfLiteInt64: in GreaterEval()
241 case kTfLiteInt64: in GreaterEqualEval()
273 case kTfLiteInt64: in LessEval()
305 case kTfLiteInt64: in LessEqualEval()
|
| D | floor_mod.cc | 84 if (type != kTfLiteInt32 && type != kTfLiteFloat32 && type != kTfLiteInt64) { in Prepare()
110 if (input2->type == kTfLiteInt32 || input2->type == kTfLiteInt64) { in EvalImpl()
147 case kTfLiteInt64: { in Eval()
|
| D | reverse.cc | 43 input->type != kTfLiteInt64) { in Prepare()
100 case kTfLiteInt64: { in Eval()
|
| D | cast.cc | 75 case kTfLiteInt64: in copyToTensor()
107 case kTfLiteInt64: in Eval()
|
| D | space_to_depth.cc | 54 data_type == kTfLiteInt64); in Prepare()
118 case kTfLiteInt64: in Eval()
|
| D | unique.cc | 108 case kTfLiteInt64: in EvalImpl()
137 case kTfLiteInt64: in Eval()
|
| D | pack.cc | 43 input0->type != kTfLiteInt16 && input0->type != kTfLiteInt64) { in Prepare()
117 case kTfLiteInt64: { in Eval()
|
| D | reduce.cc | 187 temp_sum->type = kTfLiteInt64; in InitializeTemporaries()
189 case kTfLiteInt64: in InitializeTemporaries()
190 temp_sum->type = kTfLiteInt64; in InitializeTemporaries()
364 case kTfLiteInt64: in EvalMean()
504 case kTfLiteInt64: in EvalGeneric()
|
| D | concatenation.cc | 62 input_type == kTfLiteInt32 || input_type == kTfLiteInt64); in Prepare()
170 case kTfLiteInt64: in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 71 case kTfLiteInt64: in GetTensorFlowDataType()
95 return kTfLiteInt64; in GetTensorFlowLiteType()
|
| D | util_test.cc | 115 EXPECT_EQ(TF_INT64, GetTensorFlowDataType(kTfLiteInt64)); in TEST()
126 EXPECT_EQ(kTfLiteInt64, GetTensorFlowLiteType(TF_INT64)); in TEST()
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | tflite_driver.cc | 183 case kTfLiteInt64: in Check()
287 case kTfLiteInt64: { in SetInput()
329 case kTfLiteInt64: in SetExpectation()
397 case kTfLiteInt64: in ReadOutput()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | tflite_driver.cc | 183 case kTfLiteInt64: in Check()
286 case kTfLiteInt64: { in SetInput()
335 case kTfLiteInt64: in SetExpectation()
400 case kTfLiteInt64: in ReadOutput()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | python_utils.cc | 43 case kTfLiteInt64: in TfLiteTypeToPyArrayType()
72 return kTfLiteInt64; in TfLiteTypeFromPyArray()
|
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | tensor_jni.cc | 82 case kTfLiteInt64: in ElementByteSize()
117 case kTfLiteInt64: { in WriteOneDimensionalArray()
165 case kTfLiteInt64: { in ReadOneDimensionalArray()
|