Searched refs:CheckNumericsOp (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | check_numerics_op.cc | 76 class CheckNumericsOp;
82 class CheckNumericsOp<CPUDevice, T> : public OpKernel { class
84 explicit CheckNumericsOp(OpKernelConstruction* context) : OpKernel(context) { in CheckNumericsOp() function in tensorflow::__anon5c1c7c0c0111::CheckNumericsOp
150 class CheckNumericsV2Op<CPUDevice, T> : public CheckNumericsOp<CPUDevice, T> {
153 : CheckNumericsOp<CPUDevice, T>(context) {} in CheckNumericsV2Op()
195 class CheckNumericsOp<GPUDevice, T> : public AsyncOpKernel { class
199 explicit CheckNumericsOp(OpKernelConstruction* context) in CheckNumericsOp() function in tensorflow::__anon5c1c7c0c0111::CheckNumericsOp
312 class CheckNumericsV2Op<GPUDevice, T> : public CheckNumericsOp<GPUDevice, T> {
315 : CheckNumericsOp<GPUDevice, T>(context) {} in CheckNumericsV2Op()
366 CheckNumericsOp<CPUDevice, T>);
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | check_numerics_op.cc | 24 class CheckNumericsOp : public XlaOpKernel { class
26 explicit CheckNumericsOp(OpKernelConstruction* context) in CheckNumericsOp() function in tensorflow::__anonaedefa7e0111::CheckNumericsOp
44 TF_DISALLOW_COPY_AND_ASSIGN(CheckNumericsOp);
47 REGISTER_XLA_OP(Name("CheckNumerics"), CheckNumericsOp);
|