Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.h105 XLA_HloModule ToC(const xla::HloModule& module);
107 const XLA_HloModule& c_module);
108 void Free(XLA_HloModule* c_module);
Dc_api_decl.h301 typedef struct XLA_HloModule { struct
304 } XLA_HloModule; argument
Dtpu_executor_c_api.h289 Tpu_Compiler* compiler, XLA_HloModule* se_hlo_module,
291 XLA_HloModule* result, TF_Status* status);
294 Tpu_Compiler* compiler, XLA_HloModule* se_hlo_module,
331 TFTPU_CAPI_EXPORT XLA_HloModule
Dc_api_conversions.cc380 XLA_HloModule ToC(const xla::HloModule& module) { in ToC()
381 XLA_HloModule c_module; in ToC()
388 const XLA_HloModule& c_module) { in FromC()
396 void Free(XLA_HloModule* c_module) { in Free()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc222 XLA_HloModule hlo_module; in RunHloPasses()
230 XLA_HloModule result; in RunHloPasses()
250 XLA_HloModule hlo_module; in RunBackend()
331 XLA_HloModule c_module = in Compile()