Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_format.h247 inline int GetFilterTensorSpatialDimIndex(int num_dims, in GetFilterTensorSpatialDimIndex() function
563 dim_sizes[GetFilterTensorSpatialDimIndex(dims, format, dim)] = spatial[dim]; in ShapeFromFilterTensorFormat()
H A Dtensor_format_test.cc199 EXPECT_EQ(GetFilterTensorSpatialDimIndex(num_dims, format, i), in RunDimensionIndexesTest()