Searched refs:GetShapeDebugString (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 438 const std::string GetShapeDebugString(const TfLiteIntArray* shape) { in GetShapeDebugString() function
472 GetShapeDebugString(input1->dims).c_str(), in CalculateShapeForBroadcast()
473 GetShapeDebugString(input2->dims).c_str()); in CalculateShapeForBroadcast()
510 GetShapeDebugString(input1->dims).c_str(), in CalculateShapeForBroadcast()
511 GetShapeDebugString(input2->dims).c_str(), in CalculateShapeForBroadcast()
512 GetShapeDebugString(input3->dims).c_str()); in CalculateShapeForBroadcast()
|
| D | kernel_util_test.cc | 334 TEST_F(KernelUtilTest, GetShapeDebugString) { in TEST_F() argument
336 EXPECT_EQ("[]", GetShapeDebugString(dims0)); in TEST_F()
341 EXPECT_EQ("[1]", GetShapeDebugString(dims1)); in TEST_F()
347 EXPECT_EQ("[2,3]", GetShapeDebugString(dims2)); in TEST_F()
354 EXPECT_EQ("[4,5,6]", GetShapeDebugString(dims3)); in TEST_F()
|
| D | kernel_util.h | 300 const std::string GetShapeDebugString(const TfLiteIntArray* shape);
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel_test.cc | 492 TEST(ValidateOutputTensorShapeConsistencyTest, GetShapeDebugString) { in TEST() argument
495 EXPECT_EQ("[1]", GetShapeDebugString(dims1)); in TEST()
501 EXPECT_EQ("[2,3]", GetShapeDebugString(dims2)); in TEST()
508 EXPECT_EQ("[4,5,6]", GetShapeDebugString(dims3)); in TEST()
|
| D | kernel.cc | 679 GetShapeDebugString(tfl_tensor->dims); in ValidateOutputTensorShapeConsistency()
|
| /external/tensorflow/tensorflow/lite/ |
| D | tensorflow_profiler_logger.cc | 96 std::string dims = tensor->dims ? GetShapeDebugString(tensor->dims) : "[]"; in AddTraceMe()
|