| /external/tensorflow/tensorflow/lite/kernels/ |
| D | arg_min_max.cc | 42 if (axis->type == kTfLiteInt64) { in ResizeOutput()
75 axis->type == kTfLiteInt32 || axis->type == kTfLiteInt64); in Prepare()
86 case kTfLiteInt64: in Prepare()
87 output->type = kTfLiteInt64; in Prepare()
173 case kTfLiteInt64: { in Eval()
225 case kTfLiteInt64: { in Eval()
|
| D | shape.cc | 54 case kTfLiteInt64: in Prepare()
55 output->type = kTfLiteInt64; in Prepare()
83 case kTfLiteInt64: in Prepare()
|
| D | reverse_sequence.cc | 48 input->type != kTfLiteInt64) { in Prepare()
55 if (seq_lengths->type != kTfLiteInt32 && seq_lengths->type != kTfLiteInt64) { in Prepare()
117 case kTfLiteInt64: { in ReverseSequenceHelper()
149 case kTfLiteInt64: { in Eval()
|
| D | sparse_to_dense.cc | 133 } else if (output_shape->type == kTfLiteInt64) { in ResizeOutputShape()
173 context, indices->type == kTfLiteInt32 || indices->type == kTfLiteInt64); in Prepare()
175 output_shape->type == kTfLiteInt64); in Prepare()
177 values->type == kTfLiteInt64 || in Prepare()
244 case kTfLiteInt64: { in EvalForIndexType()
269 case kTfLiteInt64: in Eval()
|
| D | gather.cc | 51 case kTfLiteInt64: in Prepare()
70 case kTfLiteInt64: in Prepare()
161 case kTfLiteInt64: in Eval()
173 if (positions->type == kTfLiteInt64) { in Eval()
185 case kTfLiteInt64: in Eval()
|
| D | gather_nd.cc | 49 case kTfLiteInt64: in Prepare()
60 case kTfLiteInt64: in Prepare()
137 case kTfLiteInt64: in EvalGatherNd()
161 case kTfLiteInt64: in Eval()
|
| D | slice.cc | 96 } else if (begin->type == kTfLiteInt64) { in ResizeOutputShape()
129 begin->type == kTfLiteInt32 || begin->type == kTfLiteInt64); in Prepare()
131 size->type == kTfLiteInt32 || size->type == kTfLiteInt64); in Prepare()
178 } else if (begin->type == kTfLiteInt64) { in Eval()
221 case kTfLiteInt64: in Eval()
|
| D | one_hot.cc | 99 if (op_context.indices->type == kTfLiteInt64) { in OneHotCompute()
132 case kTfLiteInt64: in Prepare()
145 op_context.indices->type == kTfLiteInt64); in Prepare()
177 case kTfLiteInt64: in Eval()
|
| D | tile.cc | 70 case kTfLiteInt64: in ResizeOutput()
181 case kTfLiteInt64: in Tile()
201 case kTfLiteInt64: in TileString()
228 if (multipliers->type != kTfLiteInt32 && multipliers->type != kTfLiteInt64) { in Prepare()
267 case kTfLiteInt64: in Eval()
|
| D | floor_mod.cc | 74 if (type != kTfLiteInt32 && type != kTfLiteFloat32 && type != kTfLiteInt64) { in Prepare()
100 if (input2->type == kTfLiteInt32 || input2->type == kTfLiteInt64) { in EvalImpl()
144 case kTfLiteInt64: { in Eval()
|
| D | fill.cc | 57 case kTfLiteInt64: in ResizeOutput()
85 TF_LITE_ENSURE(context, dtype == kTfLiteInt32 || dtype == kTfLiteInt64); in Prepare()
152 case kTfLiteInt64: in Eval()
|
| D | comparisons.cc | 177 case kTfLiteInt64: in EqualEval()
227 case kTfLiteInt64: in NotEqualEval()
273 case kTfLiteInt64: in GreaterEval()
314 case kTfLiteInt64: in GreaterEqualEval()
355 case kTfLiteInt64: in LessEval()
396 case kTfLiteInt64: in LessEqualEval()
|
| D | cumsum.cc | 42 input->type == kTfLiteInt64); in Prepare()
78 case kTfLiteInt64: { in Eval()
|
| D | split_v.cc | 76 } else if (size_splits->type == kTfLiteInt64) { in ResizeOutputTensors()
137 input_type == kTfLiteInt64 || input_type == kTfLiteInt8); in Prepare()
199 case kTfLiteInt64: { in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | gather.cc | 51 case kTfLiteInt64: in Prepare()
70 case kTfLiteInt64: in Prepare()
161 case kTfLiteInt64: in Eval()
173 if (positions->type == kTfLiteInt64) { in Eval()
185 case kTfLiteInt64: in Eval()
|
| D | floor_mod.cc | 61 if (type != kTfLiteInt32 && type != kTfLiteFloat32 && type != kTfLiteInt64) { in Prepare()
77 if (input2->type == kTfLiteInt32 || input2->type == kTfLiteInt64) { in EvalImpl()
121 case kTfLiteInt64: { in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable_ops_test.cc | 93 case kTfLiteInt64: in ConvertTfLiteType()
382 interpreter_->SetTensorParametersReadWrite(kSizeTensorId, kTfLiteInt64, "", in AddTensors()
415 interpreter_->SetTensorParametersReadWrite(kSizeTwoTensorId, kTfLiteInt64, in AddTensors()
564 kTfLiteInt64, kTfLiteString, in TEST()
573 kTfLiteString, kTfLiteInt64, in TEST()
581 HashtableGraph<std::string, std::int64_t> graph(kTfLiteString, kTfLiteInt64); in TEST()
593 HashtableGraph<std::string, std::int64_t> graph(kTfLiteString, kTfLiteInt64); in TEST()
607 HashtableGraph<std::string, std::int64_t> graph(kTfLiteString, kTfLiteInt64); in TEST()
625 HashtableGraph<std::string, std::int64_t> graph(kTfLiteString, kTfLiteInt64); in TEST()
679 EXPECT_TRUE(hashtable->GetKeyType() == kTfLiteInt64); in TEST()
[all …]
|
| D | hashtable_import.cc | 50 TF_LITE_ENSURE(context, (key_tensor->type == kTfLiteInt64 && in PrepareHashtableImport()
53 value_tensor->type == kTfLiteInt64)); in PrepareHashtableImport()
|
| D | hashtable_find.cc | 57 TF_LITE_ENSURE(context, (key_tensor->type == kTfLiteInt64 && in PrepareHashtableFind()
60 output_tensor->type == kTfLiteInt64)); in PrepareHashtableFind()
|
| D | hashtable.cc | 78 TF_LITE_ENSURE(context, (params->key_dtype == kTfLiteInt64 && in PrepareHashtable()
81 params->value_dtype == kTfLiteInt64)); in PrepareHashtable()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_tensor_to_tensor_tflite.cc | 92 case kTfLiteInt64: in GetFirstDimensionSize()
130 case kTfLiteInt64: { in TensorShapeFromTensor()
165 case kTfLiteInt64: in GetMaxWidthValueRowID()
196 case kTfLiteInt64: in GetMaxWidthRowSplit()
347 case kTfLiteInt64: in CalculateOutputIndexValueRowID()
392 case kTfLiteInt64: in CalculateOutputIndexRowSplit()
535 case kTfLiteInt64: in SetOutput()
599 if (input_shape.type != kTfLiteInt32 && input_shape.type != kTfLiteInt64) { in Prepare()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 77 case kTfLiteInt64: in GetTensorFlowDataType()
113 return kTfLiteInt64; in GetTensorFlowLiteType()
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | static_hashtable.cc | 85 if (key_type == kTfLiteInt64 && value_type == kTfLiteString) { in CreateStaticHashtable()
87 } else if (key_type == kTfLiteString && value_type == kTfLiteInt64) { in CreateStaticHashtable()
|
| /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/python/interpreter_wrapper/ |
| D | numpy.cc | 53 case kTfLiteInt64: in TfLiteTypeToPyArrayType()
94 return kTfLiteInt64; in TfLiteTypeFromPyType()
|