Searched refs:infeed_buffers (Results 1 – 1 of 1) sorted by relevance
36 ShapeTree<InfeedBuffer> infeed_buffers = in ExecuteOnStream() local48 ShapeUtil::Equal(infeed_buffers.shape(), infeed_shape.tuple_shapes(0))) in ExecuteOnStream()52 << ShapeUtil::HumanStringWithLayout(infeed_buffers.shape()); in ExecuteOnStream()60 const Shape& shape = ShapeUtil::GetSubshape(infeed_buffers.shape(), in ExecuteOnStream()70 infeed_buffers.mutable_element(ShapeIndexView(index, 1)); in ExecuteOnStream()