Home
last modified time | relevance | path

Searched refs:INPUT_CHANNELS (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dshape_test.cc42 EXPECT_EQ(1, OIHW.get(Axis::INPUT_CHANNELS)); in TEST()
54 std::vector<Axis> expected = {Axis::OUTPUT_CHANNELS, Axis::INPUT_CHANNELS, in TEST()
84 std::vector<Axis> expected = {Axis::OUTPUT_CHANNELS, Axis::INPUT_CHANNELS, in TEST()
Dshape.h35 INPUT_CHANNELS = 2, enumerator
235 TFLITE_GPU_AXIS_TRAITS(INPUT_CHANNELS, i);
357 Axis::INPUT_CHANNELS);
358 TFLITE_GPU_LAYOUT_TRAITS(OIHW, Axis::OUTPUT_CHANNELS, Axis::INPUT_CHANNELS,
360 TFLITE_GPU_LAYOUT_TRAITS(IOHW, Axis::INPUT_CHANNELS, Axis::OUTPUT_CHANNELS,
362 TFLITE_GPU_LAYOUT_TRAITS(IHWO, Axis::INPUT_CHANNELS, Axis::HEIGHT, Axis::WIDTH,
375 Axis::WIDTH, Axis::DEPTH, Axis::INPUT_CHANNELS);
Dshape.cc60 case Axis::INPUT_CHANNELS: in ToString()
Doperations.cc537 attr.weights.shape.get<Axis::INPUT_CHANNELS>()); in CalculateOutputShape()
546 attr.weights.shape.get<Axis::INPUT_CHANNELS>()); in CalculateOutputShape()