Home
last modified time | relevance | path

Searched refs:XLA_HloModuleConfig (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h276 typedef struct XLA_HloModuleConfig { struct
287 } XLA_HloModuleConfig; argument
298 XLA_HloModuleConfig* module_config;
303 XLA_HloModuleConfig module_config;
Dc_api_conversions.h111 XLA_HloModuleConfig ToC(const xla::HloModuleConfig& config);
112 xla::HloModuleConfig FromC(const XLA_HloModuleConfig& c_config);
113 void Free(XLA_HloModuleConfig* c_config);
Dc_api_conversions.cc402 const XLA_HloModuleConfig& se_config) { in ConfigWithLayout()
413 XLA_HloModuleConfig ToC(const xla::HloModuleConfig& config) { in ToC()
414 XLA_HloModuleConfig hlo_config; in ToC()
453 xla::HloModuleConfig FromC(const XLA_HloModuleConfig& c_config) { in FromC()
475 void Free(XLA_HloModuleConfig* c_config) { in Free()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc283 new XLA_HloModuleConfig[module_group->size()]; in Compile()