Searched refs:GetFilterTensorInnerInputChannelsDimIndex (Results 1 – 2 of 2) sorted by relevance
264 inline int GetFilterTensorInnerInputChannelsDimIndex( in GetFilterTensorInnerInputChannelsDimIndex() function570 dim_sizes[GetFilterTensorInnerInputChannelsDimIndex(dims, format)] = 4; in ShapeFromFilterTensorFormat()
663 GetFilterTensorInnerInputChannelsDimIndex(rank, filter_format)), in Conv2DShapeImpl()