Searched refs:CopyShapeAndType (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util_test.cc | 75 TEST(UtilTest, CopyShapeAndType) { in TEST() argument
82 EXPECT_EQ(CopyShapeAndType(&context, Tensor(), &dst), kTfLiteOk); in TEST()
86 EXPECT_EQ(CopyShapeAndType(&context, Tensor(DT_FLOAT, {1, 2}), &dst), in TEST()
91 EXPECT_EQ(CopyShapeAndType(&context, Tensor(DT_INT32, {1, 2}), &dst), in TEST()
96 EXPECT_EQ(CopyShapeAndType(&context, Tensor(DT_FLOAT, {1LL << 44, 2}), &dst), in TEST()
103 CopyShapeAndType(&context, Tensor(tensorflow::DT_HALF, {1, 2}), &dst), in TEST()
|
| D | util.h | 34 TfLiteStatus CopyShapeAndType(TfLiteContext* context,
|
| D | util.cc | 29 TfLiteStatus CopyShapeAndType(TfLiteContext* context, in CopyShapeAndType() function
|
| D | kernel.cc | 634 TF_LITE_ENSURE_OK(context, CopyShapeAndType(context, tf_tensor, tensor)); in Eval()
|