Searched refs:GetTensorDimsFromSpatialDims (Results 1 – 4 of 4) sorted by relevance
144 inline int GetTensorDimsFromSpatialDims(int num_spatial_dims, in GetTensorDimsFromSpatialDims() function524 const int dims = GetTensorDimsFromSpatialDims(spatial.size(), format); in ShapeFromFormat()
175 int num_dims = GetTensorDimsFromSpatialDims(num_spatial_dims, format); in RunDimensionIndexesTest()
498 const int num_dims = GetTensorDimsFromSpatialDims(spatial.size(), format); in MakeShapeFromFormat()524 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format); in DimensionsFromShape()548 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format); in ShapeFromDimensions()599 const int rank = GetTensorDimsFromSpatialDims(num_spatial_dims, data_format); in Conv2DShapeImpl()
2482 GetTensorDimsFromSpatialDims(num_spatial_dims, data_format); in __anondb9326b23702()2536 GetTensorDimsFromSpatialDims(num_spatial_dims, data_format); in __anondb9326b23802()