Searched defs:indexed_shape (Results 1 – 6 of 6) sorted by relevance
37 for (const ShapeUtil::IndexedShape& indexed_shape : in Run() local48 for (const ShapeUtil::IndexedShape& indexed_shape : in Run() local
342 for (const auto& indexed_shape : ShapeUtil::GetLeafShapes(infeed->shape())) { in HandleInfeed() local358 for (const auto& indexed_shape : in HandleOutfeed() local934 for (const auto& indexed_shape : in HandleFusion() local1134 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesRead() local1152 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesWritten() local
1613 for (auto& indexed_shape : ShapeUtil::GetLeafShapes(instruction->shape())) { in GetInPlaceInputOutputPairs() local
1981 for (const auto& indexed_shape : in AddInputAndOutputRequiredAssignments() local
562 const Shape& indexed_shape = use->operand(0)->shape(); in CreateLiteralForConstrainedUses() local
205 for (const ShapeUtil::IndexedShape& indexed_shape : in FromLiterals() local