| /external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/ |
| D | predict.cc | 83 TF_LITE_ENSURE_EQ(context, lookup->type, kTfLiteInt32); in Prepare()
84 TF_LITE_ENSURE_EQ(context, model_key->type, kTfLiteInt32); in Prepare()
85 TF_LITE_ENSURE_EQ(context, model_label->type, kTfLiteInt32); in Prepare()
101 TF_LITE_ENSURE_EQ(context, output_label->type, kTfLiteInt32); in Prepare()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | gather.cc | 42 TF_LITE_ENSURE_EQ(context, positions->type, kTfLiteInt32); in Prepare()
53 case kTfLiteInt32: { in Prepare()
100 case kTfLiteInt32: in Eval()
|
| D | strided_slice.cc | 155 TF_LITE_ENSURE_EQ(context, op_context.begin->type, kTfLiteInt32); in Prepare()
156 TF_LITE_ENSURE_EQ(context, op_context.end->type, kTfLiteInt32); in Prepare()
157 TF_LITE_ENSURE_EQ(context, op_context.strides->type, kTfLiteInt32); in Prepare()
223 case kTfLiteInt32: in Eval()
|
| D | embedding_lookup_sparse.cc | 86 TF_LITE_ENSURE_EQ(context, ids->type, kTfLiteInt32); in Prepare()
90 TF_LITE_ENSURE_EQ(context, indices->type, kTfLiteInt32); in Prepare()
94 TF_LITE_ENSURE_EQ(context, shape->type, kTfLiteInt32); in Prepare()
|
| D | mean.cc | 141 scratch_tensor->type = kTfLiteInt32; in InitializeTemporaries()
151 resolved_axis->type = kTfLiteInt32; in InitializeTemporaries()
203 case kTfLiteInt32: in Eval()
|
| D | space_to_depth.cc | 53 data_type == kTfLiteInt32 || data_type == kTfLiteInt64); in Prepare()
101 case kTfLiteInt32: in Eval()
|
| D | topk_v2.cc | 35 TF_LITE_ENSURE_EQ(context, top_k->type, kTfLiteInt32); in ResizeOutput()
170 TF_LITE_ENSURE_EQ(context, top_k->type, kTfLiteInt32); in Prepare()
209 case kTfLiteInt32: in Eval()
|
| D | hashtable_lookup.cc | 65 TF_LITE_ENSURE_EQ(context, lookup->type, kTfLiteInt32); in Prepare()
69 TF_LITE_ENSURE_EQ(context, key->type, kTfLiteInt32); in Prepare()
|
| D | embedding_lookup.cc | 56 TF_LITE_ENSURE_EQ(context, lookup->type, kTfLiteInt32); in Prepare()
|
| D | resize_bilinear.cc | 64 TF_LITE_ENSURE_EQ(context, size->type, kTfLiteInt32); in Prepare()
|
| D | transpose.cc | 127 case kTfLiteInt32: in Eval()
|
| D | pad.cc | 143 case kTfLiteInt32: in Eval()
|
| D | space_to_batch_nd.cc | 140 case kTfLiteInt32: in Eval()
|
| D | batch_to_space_nd.cc | 143 case kTfLiteInt32: in Eval()
|
| D | depthwise_conv.cc | 111 TF_LITE_ENSURE_EQ(context, bias->type, kTfLiteInt32); in Prepare()
|
| D | conv.cc | 143 TF_LITE_ENSURE_EQ(context, bias->type, kTfLiteInt32); in Prepare()
|
| /external/tensorflow/tensorflow/contrib/lite/testing/ |
| D | tflite_driver.cc | 72 case kTfLiteInt32: in Check()
178 case kTfLiteInt32: { in SetInput()
215 case kTfLiteInt32: in SetExpectation()
|
| /external/tensorflow/tensorflow/contrib/lite/java/src/main/native/ |
| D | tensor_jni.cc | 51 case kTfLiteInt32: { in writeOneDimensionalArray()
102 case kTfLiteInt32: { in readOneDimensionalArray()
157 case kTfLiteInt32: in elementByteSize()
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| D | optional_debug_tools.cc | 43 case kTfLiteInt32: in TensorTypeName()
|
| D | interpreter_test.cc | 108 {kTfLiteInt32, sizeof(int32_t)}, in TEST()
142 {kTfLiteInt32, sizeof(int32_t), reinterpret_cast<const char*>(int32s)}, in TEST()
181 {kTfLiteInt32}, in TEST()
429 ASSERT_EQ(interpreter.SetTensorParametersReadWrite(2, kTfLiteInt32, "", {1}, in TEST()
435 ASSERT_EQ(interpreter.SetTensorParametersReadWrite(4, kTfLiteInt32, "", {1}, in TEST()
|
| D | context.h | 130 kTfLiteInt32 = 2, enumerator
|
| D | interpreter.h | 37 return kTfLiteInt32;
|
| D | nnapi_delegate.cc | 99 case kTfLiteInt32: in addTensorOperands()
|
| D | interpreter.cc | 222 case kTfLiteInt32: in BytesRequired()
|
| /external/tensorflow/tensorflow/contrib/lite/examples/label_image/ |
| D | bitmap_helpers_impl.h | 58 interpreter->SetTensorParametersReadWrite(1, kTfLiteInt32, "new_size", {2}, in resize()
|