Searched refs:spatial_dim_indices (Results 1 – 1 of 1) sorted by relevance
1091 ArrayRef<int64_t> spatial_dim_indices, tensorflow::TensorFormat format, in GetConvDimensionNumbersAttr() argument1093 int64_t num_spatial_dims = spatial_dim_indices.size(); in GetConvDimensionNumbersAttr()1101 GetI64ElementsAttr(spatial_dim_indices, builder); in GetConvDimensionNumbersAttr()1192 SmallVector<int64_t, num_spatial_dims> spatial_dim_indices; in matchAndRewrite() local1204 spatial_dim_indices.push_back(dim); in matchAndRewrite()1237 spatial_dim_indices, data_format, &rewriter); in matchAndRewrite()