Searched refs:use_tpu_host_allocator_for_inputs_ (Results 1 – 1 of 1) sorted by relevance
1646 use_tpu_host_allocator_for_inputs_ = in TfToTfrtConversionPass()1675 tpu_transfer_result_to_host_, use_tpu_host_allocator_for_inputs_}, in runOnFunction()1897 Option<bool> use_tpu_host_allocator_for_inputs_{ member in tensorflow::__anon42f45b140111::TfToTfrtConversionPass