Searched refs:DimensionSizes (Results 1 – 4 of 4) sorted by relevance
119 std::vector<int64> DimensionSizes() const;
474 std::vector<int64> XlaCompiler::Argument::DimensionSizes() const { in DimensionSizes() function in tensorflow::XlaCompiler::Argument1159 xla::Reshape(arg_handles[i], arg.DimensionSizes()), arg.type); in BuildArguments()
209 ArrayRef<const SCEV *> DimensionSizes, MemoryKind Kind,271 return DimensionSizes.size(); in getNumberOfDimensions()281 return DimensionSizes[Dim]; in getDimensionSize()400 SmallVector<const SCEV *, 4> DimensionSizes; variable
324 int SharedDims = std::min(NewSizes.size(), DimensionSizes.size()); in updateSizes()326 int ExtraDimsOld = DimensionSizes.size() - SharedDims; in updateSizes()331 auto *KnownSize = DimensionSizes[i + ExtraDimsOld]; in updateSizes()336 if (DimensionSizes.size() >= NewSizes.size()) in updateSizes()340 DimensionSizes.clear(); in updateSizes()341 DimensionSizes.insert(DimensionSizes.begin(), NewSizes.begin(), in updateSizes()344 for (const SCEV *Expr : DimensionSizes) { in updateSizes()