Crate caffe2op_enforce

source ·

Structs

  • EnforceFiniteOp
    | Raise if there is NaN or Inf values in | the input tensor. |
  • EnsureCPUOutputOp
    | This Op always create TensorCPU output, | and may involves cross-device MemCpy. | | Under CPU Context, this Op takes TensorCPU | as input. Under the CUDA Context, this | Op accepts either CUDA or CPU Tensor | input. |
  • EnsureClippedOp
    | Given a tensor, apply clip after gradient | is applied; when the param is sparse | as indicated by valid indices and grad, | in-place is required |