Home
last modified time | relevance | path

Searched refs:IsBatchedWidth (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dtensor_desc.cc276 if (HasAxis(Axis::BATCH) && !IsBatchedWidth()) { in GetLinkingContextFromWriteSelector()
516 if (HasAxis(Axis::BATCH) && !IsBatchedWidth()) { in PerformGetWHOffsetSelector()
716 if (layout == Layout::HWC || (IsBatchedWidth() && layout == Layout::BHWC)) { in GetPhysicalCoords()
721 (IsBatchedWidth() && layout == Layout::BHWDC)) { in GetPhysicalCoords()
787 if (IsBatchedWidth()) { in GetSliceStrideSize()
835 if (HasAxis(Axis::BATCH) && !IsBatchedWidth()) { in ParseCoordsFromArgs()
850 bool TensorDescriptor::IsBatchedWidth() const { in IsBatchedWidth() function in tflite::gpu::TensorDescriptor
Dtensor_desc.h145 bool IsBatchedWidth() const;