Searched refs:use_xla (Results 1 – 2 of 2) sorted by relevance
323 def _GetVal(use_xla): argument328 if use_xla:340 gpu_value = _GetVal(use_xla=True)341 cpu_value = _GetVal(use_xla=False)363 def _GetVal(use_xla): argument372 if use_xla:396 gpu_value = _GetVal(use_xla=True)397 cpu_value = _GetVal(use_xla=False)
65 void SetUseXla(bool use_xla) { use_xla_ = use_xla; } in SetUseXla() argument