Crate caffe2op_enforce
source ·Structs
- | Raise if there is NaN or Inf values in | the input tensor. |
- | 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. |
- | 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 |