Searched refs:infeed_instruction_shape (Results 1 – 1 of 1) sorted by relevance
1287 const Shape infeed_instruction_shape = in Infeed() local1289 *instr.mutable_shape() = infeed_instruction_shape.ToProto(); in Infeed()1363 const Shape infeed_instruction_shape = in InfeedWithToken() local1365 *instr.mutable_shape() = infeed_instruction_shape.ToProto(); in InfeedWithToken()