Searched refs:abnormal_detected (Results 1 – 2 of 2) sorted by relevance
40 int abnormal_detected[2]) { in CheckNumericsKernel()48 abnormal_detected[0] = 1; in CheckNumericsKernel()51 abnormal_detected[1] = 1; in CheckNumericsKernel()63 int abnormal_detected[3]) { in CheckNumericsKernelV2()71 abnormal_detected[0] = 1; in CheckNumericsKernelV2()74 abnormal_detected[data[offset] < static_cast<T>(0.f) ? 1 : 2] = 1; in CheckNumericsKernelV2()87 int abnormal_detected[2]) { in Run()94 0, d.stream(), data, size, abnormal_detected)); in Run()105 int abnormal_detected[3]) { in Run()113 abnormal_detected)); in Run()
50 int abnormal_detected[2]);60 int abnormal_detected[3]);216 Tensor abnormal_detected; in ComputeAsync() local220 &abnormal_detected)); in ComputeAsync()227 abnormal_detected.flat<int>().data(), in ComputeAsync()228 abnormal_detected.flat<int>().size()); in ComputeAsync()230 abnormal_detected.flat<int>().size() * sizeof(int)); in ComputeAsync()235 abnormal_detected.flat<int>().data()); in ComputeAsync()258 TensorReference abnormal_detected_ref(abnormal_detected); in ComputeAsync()282 int* abnormal_detected) { in RunKernel() argument[all …]