MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / ToChannelOrder

Function ToChannelOrder

tensorflow/lite/delegates/gpu/cl/cl_image_format.cc:22–35  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

20namespace cl {
21
22cl_channel_order ToChannelOrder(int num_channels) {
23 switch (num_channels) {
24 case 1:
25 return CL_R;
26 case 2:
27 return CL_RG;
28 case 3:
29 return CL_RGB;
30 case 4:
31 return CL_RGBA;
32 default:
33 return -1;
34 }
35}
36
37cl_channel_type ToImageChannelType(DataType data_type) {
38 switch (data_type) {

Callers 2

AllocateTensorMemoryFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected