Searched refs:literal_subshape (Results 1 – 2 of 2) sorted by relevance
139 shape_, [&](const Shape& literal_subshape, const ShapeIndex& index) { in SetLiteral() argument140 if (!literal_subshape.IsTuple()) { in SetLiteral()141 total_size_bytes += ShapeUtil::ByteSizeOf(literal_subshape, 8); in SetLiteral()
510 ForEachSubshape(shape, [&](const Shape& literal_subshape, in SubshapeCount()