Home
last modified time | relevance | path

Searched defs:CheckNumericsOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcheck_numerics_op.cc24 class CheckNumericsOp : public XlaOpKernel { class
26 explicit CheckNumericsOp(OpKernelConstruction* context) in CheckNumericsOp() function in tensorflow::__anonaedefa7e0111::CheckNumericsOp
/external/tensorflow/tensorflow/core/kernels/
Dcheck_numerics_op.cc82 class CheckNumericsOp<CPUDevice, T> : public OpKernel { class
84 explicit CheckNumericsOp(OpKernelConstruction* context) : OpKernel(context) { in CheckNumericsOp() function in tensorflow::__anon5c1c7c0c0111::CheckNumericsOp
195 class CheckNumericsOp<GPUDevice, T> : public AsyncOpKernel { class
199 explicit CheckNumericsOp(OpKernelConstruction* context) in CheckNumericsOp() function in tensorflow::__anon5c1c7c0c0111::CheckNumericsOp