Searched defs:indexed_shape (Results 1 – 6 of 6) sorted by relevance
40 for (const ShapeUtil::IndexedShape& indexed_shape : in Run() local51 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() local963 for (const auto& indexed_shape : in HandleFusion() local1191 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesWritten() local
2458 for (const auto& indexed_shape : in AddInputAndOutputRequiredAssignments() local
622 const Shape& indexed_shape = use->operand(0)->shape(); in CreateLiteralForConstrainedUses() local
215 for (const ShapeUtil::IndexedShape& indexed_shape : in FromLiterals() local
423 auto indexed_shape = get_indexed_shape(self_, List<std::optional<Tensor>>(indices_)); local