Searched refs:set_infeed_config (Results 1 – 5 of 5) sorted by relevance
1513 void set_infeed_config(const string& config);
971 void set_infeed_config(const string& config) { infeed_config_ = config; } in set_infeed_config() function
3400 void HloInstruction::set_infeed_config(const string& config) { in set_infeed_config() function in xla::HloInstruction3401 return Cast<HloInfeedInstruction>(this)->set_infeed_config(config); in set_infeed_config()
1784 proto.set_infeed_config(infeed_config_); in ToProto()
1290 instr.set_infeed_config(config); in Infeed()1366 instr.set_infeed_config(config); in InfeedWithToken()