Searched defs:spatial_dim (Results 1 – 5 of 5) sorted by relevance
53 int spatial_dim[3]; member65 int spatial_dim[3]; member
216 int spatial_dim) { in GetTensorSpatialDimIndex()590 for (int spatial_dim = 0; spatial_dim < num_src_spatial_dims; ++spatial_dim) { in ShapeFromFormat() local
61 Padding padding, int64 padding_before, int64 padding_after, int spatial_dim, in ConvBackpropExtractAndVerifyDimension()
141 for (int spatial_dim = 0; spatial_dim < num_spatial_dims; ++spatial_dim) { in XlaTensorFormat() local
344 for (int spatial_dim = 0; spatial_dim < spatial.size(); spatial_dim++) { in MakeShapeFromFormat() local