Home
last modified time | relevance | path

Searched refs:launch_ok (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.cc180 bool launch_ok; in ExecuteOnStream() local
187 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
195 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
203 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
204 if (launch_ok) { in ExecuteOnStream()
205 launch_ok = stream in ExecuteOnStream()
217 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
218 if (launch_ok) { in ExecuteOnStream()
219 launch_ok = in ExecuteOnStream()
232 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
[all …]
Dtriangular_solve_thunk.cc90 bool launch_ok; in ExecuteOnStream() local
98 launch_ok = stream in ExecuteOnStream()
108 launch_ok = stream in ExecuteOnStream()
118 launch_ok = in ExecuteOnStream()
129 launch_ok = in ExecuteOnStream()
141 if (!launch_ok) { in ExecuteOnStream()
Dgemm_thunk.cc273 bool launch_ok = [&]() { in RunGemm() local
308 if (!launch_ok) { in RunGemm()