Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/translate/
H A Dimport_model.cc125 pass_options.use_tpu_host_allocator_for_inputs = in ConvertTfMlirToBef()
126 options.use_tpu_host_allocator_for_inputs; in ConvertTfMlirToBef()
H A Dtfrt_compile_options.h83 bool use_tpu_host_allocator_for_inputs = false; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dtpu_passes.h56 bool use_tpu_host_allocator_for_inputs = false; member
H A Dpasses.h165 Option<bool> use_tpu_host_allocator_for_inputs{
H A Dtf_to_tfrt.cc1647 options.use_tpu_host_allocator_for_inputs; in TfToTfrtConversionPass()