Searched refs:padded_dimensions (Results 1 – 4 of 4) sorted by relevance
217 if (layout.padded_dimensions(i) < shape.dimensions(i)) { in ValidateLayoutForShape()221 i, layout.padded_dimensions(i), shape.dimensions(i)); in ValidateLayoutForShape()273 if (shape.layout().padded_dimensions(i) > shape.dimensions(i)) { in IsPadded()283 return AsInt64Slice(shape.layout().padded_dimensions()); in PaddedDimensions()289 return shape.layout().padded_dimensions(index); in PaddedDimension()
93 if (!ContainersEqual(lhs.layout().padded_dimensions(), in CompareShapes()94 rhs.layout().padded_dimensions())) { in CompareShapes()731 tensorflow::gtl::ArraySlice<int64> padded_dimensions = in ByteSizeOfElements() local733 if (!padded_dimensions.empty()) { in ByteSizeOfElements()734 CHECK_EQ(Rank(shape), padded_dimensions.size()); in ByteSizeOfElements()736 for (int64 dimension_size : padded_dimensions) { in ByteSizeOfElements()989 Permute(permutation, shape.layout().padded_dimensions())) { in PermuteDimensions()
147 // present, the size of the padded_dimensions must equal the rank of the151 repeated int64 padded_dimensions = 2; field153 // Describes the values in the padding specified by padded_dimensions. This
64 repeated int64 padded_dimensions = 2;118 Padding is defined in the optional `padded_dimensions` and `padding_value`119 fields. The field `padded_dimensions` describes the sizes (widths) to which each120 dimension is padded. If present, the number of elements in `padded_dimensions`