Searched defs:spatial_dim (Results 1 – 6 of 6) sorted by relevance
53 int spatial_dim[3]; member65 int spatial_dim[3]; member
226 int spatial_dim) { in GetTensorSpatialDimIndex()602 for (int spatial_dim = 0; spatial_dim < num_src_spatial_dims; ++spatial_dim) { in ShapeFromFormat() local
46 int64 spatial_dim = dim_nums.input_spatial_dimensions(i); in InstructionMatchesPattern() local
56 Padding padding, int64 padding_before, int64 padding_after, int spatial_dim, in ConvBackpropExtractAndVerifyDimension()
142 for (int spatial_dim = 0; spatial_dim < num_spatial_dims; ++spatial_dim) { in XlaTensorFormat() local
599 for (int spatial_dim = 0; spatial_dim < spatial.size(); spatial_dim++) { in MakeShapeFromFormat() local