Searched refs:host_infos (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | determinant_op.cc | 245 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync() argument
247 !host_infos.empty()) { in ComputeAsync()
248 for (int i = 0; i < host_infos[0].size(); ++i) { in ComputeAsync()
253 context, host_infos[0](i) >= 0, in ComputeAsync()
255 host_infos[0].data()[i], in ComputeAsync()
392 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync() argument
394 !host_infos.empty()) { in ComputeAsync()
395 for (int i = 0; i < host_infos[0].size(); ++i) { in ComputeAsync()
400 context, host_infos[0](i) >= 0, in ComputeAsync()
402 host_infos[0].data()[i], in ComputeAsync()
|
| D | lu_op_gpu.cu.cc | 239 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync() argument
241 !host_infos.empty()) { in ComputeAsync()
242 for (int i = 0; i < host_infos[0].size(); ++i) { in ComputeAsync()
245 OP_REQUIRES_ASYNC(context, host_infos[0].data()[i] <= 0, in ComputeAsync()
|
| D | matrix_solve_op.cc | 339 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync() argument
341 !host_infos.empty()) { in ComputeAsync()
342 for (int i = 0; i < host_infos[0].size(); ++i) { in ComputeAsync()
345 OP_REQUIRES_ASYNC(context, host_infos[0].data()[i] <= 0, in ComputeAsync()
|
| D | matrix_inverse_op.cc | 245 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync() argument
247 for (const auto& host_info : host_infos) { in ComputeAsync()
|