Searched refs:tensor0 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | concatenation_test.cc | 122 auto tensor0 = {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f}; in TEST() local
127 m0.SetInput(0, tensor0); in TEST()
135 m0_negative.SetInput(0, tensor0); in TEST()
143 m1.SetInput(0, tensor0); in TEST()
151 m1_negative.SetInput(0, tensor0); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 278 Tensor tensor0; in TEST() local
279 TF_ASSERT_OK(LiteralToHostTensor(literal0, DT_FLOAT, &tensor0)); in TEST()
287 auto handle = ops::XRTAllocateFromTensor(root, {tensor0, tensor1}, in TEST()
288 {tensor0.shape(), tensor1.shape()}, in TEST()
307 Tensor tensor0; in TEST() local
308 TF_ASSERT_OK(LiteralToHostTensor(literal0, DT_FLOAT, &tensor0)); in TEST()
314 auto handle = ops::XRTAllocateFromTensor(root, {tensor0}, {tensor0.shape()}, in TEST()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 544 TfLiteTensor* tensor0 = &context->tensors[node->inputs->data[0]]; in TEST() local
547 TfLiteIntArray* newSize = TfLiteIntArrayCopy(tensor0->dims); in TEST()
561 TfLiteIntArrayCopy(tensor0->dims)); in TEST()
693 TfLiteTensor* tensor0 = &context->tensors[node->inputs->data[0]]; in TEST() local
695 TfLiteIntArray* newSize = TfLiteIntArrayCopy(tensor0->dims); in TEST()
940 TfLiteTensor* tensor0 = &context->tensors[node->inputs->data[0]]; in CopyOpRegistration() local
942 TfLiteIntArray* newSize = TfLiteIntArrayCopy(tensor0->dims); in CopyOpRegistration()
|