Searched refs:is_on_xla_device (Results 1 – 2 of 2) sorted by relevance
72 bool is_on_xla_device() const { return xla_device_metadata() != nullptr; } in is_on_xla_device() function
353 /*allocate_xla_tensors=*/platform_info_.is_on_xla_device(), in Compute()522 /*allocate_xla_tensors=*/platform_info_.is_on_xla_device(), in Compute()