Searched defs:indexed_shape (Results 1 – 5 of 5) sorted by relevance
32 for (const ShapeUtil::IndexedShape& indexed_shape : in Run() local43 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() local905 for (const auto& indexed_shape : in HandleFusion() local1105 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesRead() local1123 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesWritten() local
1229 for (auto& indexed_shape : ShapeUtil::GetLeafShapes(instruction->shape())) { in GetInPlaceInputOutputPairs() local
560 const Shape& indexed_shape = use->operand(0)->shape(); in CreateLiteralForConstrainedUses() local
204 for (const ShapeUtil::IndexedShape& indexed_shape : in FromLiterals() local