Searched refs:GetTensorDimsFromSpatialDims (Results 1 – 4 of 4) sorted by relevance
144 inline int GetTensorDimsFromSpatialDims(int num_spatial_dims, in GetTensorDimsFromSpatialDims() function525 const int dims = GetTensorDimsFromSpatialDims(spatial.size(), format); in ShapeFromFormat()
175 int num_dims = GetTensorDimsFromSpatialDims(num_spatial_dims, format); in RunDimensionIndexesTest()
501 const int num_dims = GetTensorDimsFromSpatialDims(spatial.size(), format); in MakeShapeFromFormat()528 GetTensorDimsFromSpatialDims(spatial_dims.size(), format); in DimensionsFromShape()555 GetTensorDimsFromSpatialDims(spatial_dims.size(), format); in ShapeFromDimensions()607 const int rank = GetTensorDimsFromSpatialDims(num_spatial_dims, data_format); in Conv2DShapeImpl()
2515 GetTensorDimsFromSpatialDims(num_spatial_dims, data_format); in __anon38bbb0e83702()2569 GetTensorDimsFromSpatialDims(num_spatial_dims, data_format); in __anon38bbb0e83802()