Searched refs:set_infeed_config (Results 1 – 5 of 5) sorted by relevance
1798 void set_infeed_config(const string& config);
1163 void set_infeed_config(const string& config) { infeed_config_ = config; } in set_infeed_config() function
4107 void HloInstruction::set_infeed_config(const string& config) { in set_infeed_config() function in xla::HloInstruction4108 return Cast<HloInfeedInstruction>(this)->set_infeed_config(config); in set_infeed_config()
2059 proto.set_infeed_config(infeed_config_); in ToProto()
1672 instr.set_infeed_config(config); in Infeed()1764 instr.set_infeed_config(config); in InfeedWithTokenInternal()