Searched defs:spatial_dim (Results 1 – 9 of 9) sorted by relevance
41 int64 spatial_dim; member
46 int64 spatial_dim = dim_nums.input_spatial_dimensions(i); in InstructionMatchesPattern() local
5294 for (int64 spatial_dim = 0; in SwapConvOperands() local
53 int spatial_dim[3]; member65 int spatial_dim[3]; member
227 int spatial_dim) { in GetTensorSpatialDimIndex()603 for (int spatial_dim = 0; spatial_dim < num_src_spatial_dims; ++spatial_dim) { in ShapeFromFormat() local
57 Padding padding, int64 padding_before, int64 padding_after, int spatial_dim, in ConvBackpropExtractAndVerifyDimension()
143 for (int spatial_dim = 0; spatial_dim < num_spatial_dims; ++spatial_dim) { in XlaTensorFormat() local
510 for (int spatial_dim = 0, end = spatial.size(); spatial_dim < end; in MakeShapeFromFormat() local
254 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); } in spatial_dim() function