Searched refs:GetTensorSpatialDims (Results 1 – 4 of 4) sorted by relevance
116 inline int GetTensorSpatialDims(int num_dims, TensorFormat format) { in GetTensorSpatialDims() function229 spatial_dim < GetTensorSpatialDims(num_dims, format)) in GetTensorSpatialDimIndex()424 int32_t index = (GetTensorSpatialDims(num_total_dims, format) == 3) in GetTensorDimIndex()602 GetTensorSpatialDims(src_shape.dims(), src_format); in ShapeFromFormat()
70 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format); in Compile()
128 GetTensorSpatialDims(tensor_in_shape.dims(), data_format) == 2, in PoolParameters()