Searched defs:spatial_dim (Results 1 – 12 of 12) sorted by relevance
53 int spatial_dim[3]; member65 int spatial_dim[3]; member
227 int spatial_dim) { in GetTensorSpatialDimIndex()604 for (int spatial_dim = 0; spatial_dim < num_src_spatial_dims; ++spatial_dim) { in ShapeFromFormat() local
51 int64_t spatial_dim; member
46 int64_t spatial_dim = dim_nums.input_spatial_dimensions(i); in InstructionMatchesPattern() local
244 for (int64_t spatial_dim = 0; spatial_dim < spatial_dims.size(); in ShouldSkipPadOnOperand() local
6424 for (int64_t spatial_dim = 0; in SwapConvOperands() local
299 draw, spatial_dim=2, batch_size_range=(1, 4), argument
58 int spatial_dim, int filter_spatial_dim, in ConvBackpropExtractAndVerifyDimension()
151 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
262 int64_t spatial_dim(DimIndex dim) const { in spatial_dim() function
4950 const int64_t spatial_dim = in matchAndRewrite() local