Searched defs:dimension_size (Results 1 – 10 of 10) sorted by relevance
96 const int dimension_size, M multiplier, in CopyStringMultipleTimes()117 const int dimension_size = in_dimensions.data[dimension]; in TileOneDimension() local151 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
191 def dimension_size(self, axis): member in RaggedTensorDynamicShape
233 XlaOp operand, XlaOp dimension_size, in SetDimensionSizeWithRebound()
148 int64_t dimension_size = input_shape.dimensions(axis); in ArgMinMax() local
1202 def dimension_size(x, axis): function
1687 def _has_known_value(dimension_size): argument
1976 auto dimension_size = in ConvertToDynamic() local
1652 const int64_t dimension_size = in PatternMatchReshape() local