Searched defs:spatial_dim (Results 1 – 10 of 10) sorted by relevance
64 for (auto spatial_dim : in printConvolutionDimensions() local120 int64_t spatial_dim; in parseConvolutionDimensions() local
51 int64 spatial_dim; member
46 int64_t spatial_dim = dim_nums.input_spatial_dimensions(i); in InstructionMatchesPattern() local
5607 for (int64_t 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
58 int spatial_dim, int filter_spatial_dim, in ConvBackpropExtractAndVerifyDimension()
149 for (int spatial_dim = 0; spatial_dim < num_spatial_dims; ++spatial_dim) { in XlaTensorFormat() local
513 for (int spatial_dim = 0, end = spatial.size(); spatial_dim < end; in MakeShapeFromFormat() local
268 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); } in spatial_dim() function