Home
last modified time | relevance | path

Searched refs:IsExecutionThreadIncluded (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_rematerialization.h159 bool IsExecutionThreadIncluded(
H A Dhlo_rematerialization.cc1877 if (!IsExecutionThreadIncluded(execution_threads, in CalledComputationsMemoryUsage()
1887 bool HloRematerialization::IsExecutionThreadIncluded( in IsExecutionThreadIncluded() function in xla::HloRematerialization
2126 IsExecutionThreadIncluded(execution_threads, in Run()