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()
422 const Shape& tuple_element_shape = in HandleInfeed() local430 EmitBufferPointer(buffer, tuple_element_shape); in HandleInfeed()433 XfeedKind::kInfeed, tuple_element_shape, tuple_element_address)); in HandleInfeed()533 const Shape& tuple_element_shape = in HandleOutfeed() local536 tuple_element_shape, i, MinimumAlignmentForShape(tuple_element_shape), in HandleOutfeed()539 tuple_element_shape, tuple_element)); in HandleOutfeed()