Home
last modified time | relevance | path

Searched refs:set_infeed_config (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.h1513 void set_infeed_config(const string& config);
Dhlo_instructions.h971 void set_infeed_config(const string& config) { infeed_config_ = config; } in set_infeed_config() function
Dhlo_instruction.cc3400 void HloInstruction::set_infeed_config(const string& config) { in set_infeed_config() function in xla::HloInstruction
3401 return Cast<HloInfeedInstruction>(this)->set_infeed_config(config); in set_infeed_config()
Dhlo_instructions.cc1784 proto.set_infeed_config(infeed_config_); in ToProto()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1290 instr.set_infeed_config(config); in Infeed()
1366 instr.set_infeed_config(config); in InfeedWithToken()