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