Searched defs:infeed_instruction_shape (Results 1 – 2 of 2) sorted by relevance
468 const Shape& infeed_instruction_shape, XlaOp token, const string& config) { in InfeedWithTokenInternal()
1669 const Shape infeed_instruction_shape = in Infeed() local1742 const Shape infeed_instruction_shape = in InfeedWithToken() local1761 const Shape& infeed_instruction_shape, XlaOp token, const string& config) { in InfeedWithTokenInternal()