Searched refs:OIHW (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | shape_test.cc | 27 TEST(OIHW, Smoke) { in TEST() argument
28 OIHW OIHW; in TEST() local
31 OIHW.i = 1; in TEST()
32 ASSERT_TRUE(OIHW.set<Axis::OUTPUT_CHANNELS>(2)); in TEST()
33 ASSERT_TRUE(OIHW.set(Axis::HEIGHT, 3)); in TEST()
34 ASSERT_TRUE(OIHW.set(3, 4)); in TEST()
37 ASSERT_FALSE(OIHW.set(5, 10)); in TEST()
38 ASSERT_FALSE(OIHW.set(Axis::CHANNELS, 10)); in TEST()
39 ASSERT_FALSE(OIHW.set<Axis::CHANNELS>(10)); in TEST()
42 EXPECT_EQ(1, OIHW.get(Axis::INPUT_CHANNELS)); in TEST()
[all …]
|
| D | shape.h | 54 OIHW = 6, enumerator
201 using OIHW = StrongShape<Layout::OIHW>; variable
358 TFLITE_GPU_LAYOUT_TRAITS(OIHW, Axis::OUTPUT_CHANNELS, Axis::INPUT_CHANNELS,
603 case Layout::OIHW:
604 return f.template operator()<Layout::OIHW>();
|
| D | shape.cc | 98 case Layout::OIHW: in ToString()
|