Searched defs:dimension_size (Results 1 – 8 of 8) sorted by relevance
96 const int dimension_size, M multiplier, in CopyStringMultipleTimes()110 const int dimension_size = in_dimensions.data[dimension]; in TileOneDimension() local144 const int dimension_size = in_dimensions.data[dimension]; in TileStringOneDimension() local
43 const int dimension_size = NumDimensions(input0) + 1; in Prepare() local
43 const int dimension_size = in_dimensions.dims(dimension); in TileOneDimension() local
195 def dimension_size(self, axis): member in RaggedTensorDynamicShape
1205 def dimension_size(x, axis): function
154 int64 dimension_size = input_shape.dimensions(axis); in ArgMinMax() local
1609 def _has_known_value(dimension_size): argument
1633 auto dimension_size = in ConvertToDynamic() local