Searched defs:infeed_instruction_shape (Results 1 – 2 of 2) sorted by relevance
485 const Shape& infeed_instruction_shape, XlaOp token, const string& config) { in InfeedWithTokenInternal()
1681 const Shape infeed_instruction_shape = in Infeed() local1754 const Shape infeed_instruction_shape = in InfeedWithToken() local1773 const Shape& infeed_instruction_shape, XlaOp token, const string& config) { in InfeedWithTokenInternal()