Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dshape_test.cc32 ASSERT_TRUE(OIHW.set<Axis::OUTPUT_CHANNELS>(2)); 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.h36 OUTPUT_CHANNELS = 3, enumerator
236 TFLITE_GPU_AXIS_TRAITS(OUTPUT_CHANNELS, o);
356 TFLITE_GPU_LAYOUT_TRAITS(OHWI, Axis::OUTPUT_CHANNELS, Axis::HEIGHT, Axis::WIDTH,
358 TFLITE_GPU_LAYOUT_TRAITS(OIHW, Axis::OUTPUT_CHANNELS, Axis::INPUT_CHANNELS,
360 TFLITE_GPU_LAYOUT_TRAITS(IOHW, Axis::INPUT_CHANNELS, Axis::OUTPUT_CHANNELS,
363 Axis::OUTPUT_CHANNELS);
374 TFLITE_GPU_LAYOUT_TRAITS(OHWDI, Axis::OUTPUT_CHANNELS, Axis::HEIGHT,
Dshape.cc62 case Axis::OUTPUT_CHANNELS: in ToString()
Doperations.cc506 attr.weights.shape.get<Axis::OUTPUT_CHANNELS>()); in CalculateOutputShape()
514 attr.weights.shape.get<Axis::OUTPUT_CHANNELS>()); in CalculateOutputShape()
521 attr.weights.shape.get<Axis::OUTPUT_CHANNELS>()); in CalculateOutputShape()
529 attr.weights.shape.get<Axis::OUTPUT_CHANNELS>()); in CalculateOutputShape()
536 attr.weights.shape.get<Axis::OUTPUT_CHANNELS>() * in CalculateOutputShape()
545 attr.weights.shape.get<Axis::OUTPUT_CHANNELS>() * in CalculateOutputShape()