Searched refs:_xla (Results 1 – 4 of 4) sorted by relevance
31 from . import xla_extension as _xla unknown48 ops = _xla.ops49 profiler = _xla.profiler62 return _xla.get_interpreter_client()66 return _xla.get_cpu_client(asynchronous=True)78 config = _xla.GpuAllocatorConfig()80 config.kind = _xla.GpuAllocatorConfig.Kind.DEFAULT82 config.kind = _xla.GpuAllocatorConfig.Kind.PLATFORM84 config.kind = _xla.GpuAllocatorConfig.Kind.BFC89 return _xla.get_gpu_client([all …]
82 hlo_dot_graph = xla_client._xla.hlo_module_to_dot_graph(
148 properties = xla_client._xla.hlo_module_cost_analysis(1981 dlt = xla_client._xla.buffer_to_dlpack_managed_tensor(1985 y = xla_client._xla.dlpack_managed_tensor_to_buffer(dlt, self.backend)1992 dlt = xla_client._xla.buffer_to_dlpack_managed_tensor(1996 _ = xla_client._xla.dlpack_managed_tensor_to_buffer(dlt, self.backend)2006 _ = xla_client._xla.buffer_to_dlpack_managed_tensor(2012 _ = xla_client._xla.buffer_to_dlpack_managed_tensor(2018 d1 = xla_client._xla.buffer_to_dlpack_managed_tensor(2020 d2 = xla_client._xla.buffer_to_dlpack_managed_tensor(2023 y = xla_client._xla.dlpack_managed_tensor_to_buffer(d1, self.backend)[all …]
2432 name = name + "_xla",