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

Function DataLayoutString

tensorflow/stream_executor/dnn.cc:156–172  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

154}
155
156string DataLayoutString(DataLayout layout) {
157 switch (layout) {
158 case DataLayout::kYXDepthBatch:
159 return "YXDepthBatch";
160 case DataLayout::kYXBatchDepth:
161 return "YXBatchDepth";
162 case DataLayout::kBatchYXDepth:
163 return "BatchYXDepth";
164 case DataLayout::kBatchDepthYX:
165 return "BatchDepthYX";
166 case DataLayout::kBatchDepthYX4:
167 return "BatchDepthYX4";
168 default:
169 LOG(FATAL) << "Unknown data layout " << static_cast<int32>(layout);
170 }
171 return "unknown data layout";
172}
173
174string FilterLayoutString(FilterLayout layout) {
175 switch (layout) {

Callers 4

ToStringMethod · 0.85
CudnnTensorDescriptorMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected