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::__anon119a6ba00111::CheckNumericsOp
/external/tensorflow/tensorflow/core/kernels/
Dcheck_numerics_op.cc59 class CheckNumericsOp<CPUDevice, T> : public OpKernel { class
61 explicit CheckNumericsOp(OpKernelConstruction* context) : OpKernel(context) { in CheckNumericsOp() function in tensorflow::__anon4ae96e6e0111::CheckNumericsOp
115 class CheckNumericsOp<GPUDevice, T> : public AsyncOpKernel { class
119 explicit CheckNumericsOp(OpKernelConstruction* context) in CheckNumericsOp() function in tensorflow::__anon4ae96e6e0111::CheckNumericsOp