Searched refs:tuple_element_shape (Results 1 – 2 of 2) sorted by relevance
122 const Shape& tuple_element_shape = ShapeUtil::GetSubshape(shape, {i}); in TransferLiteralToInfeed() local123 int64 tuple_element_size = GetByteSizeRequirement(tuple_element_shape); in TransferLiteralToInfeed()207 const Shape& tuple_element_shape = in TransferLiteralFromOutfeed() local209 int64 size = GetByteSizeRequirement(tuple_element_shape); in TransferLiteralFromOutfeed()
358 const Shape& tuple_element_shape = in HandleInfeed() local366 EmitBufferPointer(buffer, tuple_element_shape); in HandleInfeed()369 XfeedKind::kInfeed, tuple_element_shape, tuple_element_address)); in HandleInfeed()492 const Shape& tuple_element_shape = in HandleOutfeed() local495 tuple_element_shape, i, MinimumAlignmentForShape(tuple_element_shape), in HandleOutfeed()498 tuple_element_shape, tuple_element)); in HandleOutfeed()