Searched refs:GetTensorSpatialDims (Results 1 – 4 of 4) sorted by relevance
111 inline int GetTensorSpatialDims(int num_dims, TensorFormat format) { in GetTensorSpatialDims() function218 spatial_dim < GetTensorSpatialDims(num_dims, format)) in GetTensorSpatialDimIndex()413 int32 index = (GetTensorSpatialDims(num_total_dims, format) == 3) in GetTensorDimIndex()588 GetTensorSpatialDims(src_shape.dims(), src_format); in ShapeFromFormat()
70 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format); in Compile()
55 GetTensorSpatialDims(tensor_in_shape.dims(), data_format) == 2, in PoolParameters()