Searched refs:CheckedNarrowing (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 49 NarrowT CheckedNarrowing(const WideT& wide) { in CheckedNarrowing() function
555 &CheckedNarrowing<int64, int>); in ScopedTensorDescriptor()
557 &CheckedNarrowing<int64, int>); in ScopedTensorDescriptor()
666 &CheckedNarrowing<int64, int>); in ScopedConvolutionDescriptor()
668 &CheckedNarrowing<int64, int>); in ScopedConvolutionDescriptor()
725 &CheckedNarrowing<int64, int>); in ScopedPoolingDescriptor()
727 &CheckedNarrowing<int64, int>); in ScopedPoolingDescriptor()
729 &CheckedNarrowing<int64, int>); in ScopedPoolingDescriptor()
3611 &CheckedNarrowing<int64, int>); in DoPoolBackward()
3707 &CheckedNarrowing<int64, int>); in DoPoolBackward()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 85 NarrowT CheckedNarrowing(const WideT& wide) { in CheckedNarrowing() function
512 &CheckedNarrowing<int64, int>); in CudnnTensorDescriptor()
514 &CheckedNarrowing<int64, int>); in CudnnTensorDescriptor()
665 &CheckedNarrowing<int64, int>); in CudnnConvolutionDescriptor()
667 &CheckedNarrowing<int64, int>); in CudnnConvolutionDescriptor()
671 &CheckedNarrowing<int64, int>); in CudnnConvolutionDescriptor()
730 &CheckedNarrowing<int64, int>); in CudnnPoolingDescriptor()
732 &CheckedNarrowing<int64, int>); in CudnnPoolingDescriptor()
734 &CheckedNarrowing<int64, int>); in CudnnPoolingDescriptor()
|