Searched refs:padding64 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 654 absl::Span<const int64> padding64 = convolution_descriptor.padding(); in CudnnConvolutionDescriptor() local
666 std::transform(padding64.cbegin(), padding64.cend(), padding.begin(), in CudnnConvolutionDescriptor()
722 absl::Span<const int64> padding64 = pooling_descriptor.padding(); in CudnnPoolingDescriptor() local
731 std::transform(padding64.cbegin(), padding64.cend(), padding.begin(), in CudnnPoolingDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 656 const auto& padding64 = convolution_descriptor.padding(); in ScopedConvolutionDescriptor() local
667 std::transform(padding64.cbegin(), padding64.cend(), padding.begin(), in ScopedConvolutionDescriptor()
717 absl::Span<const int64> padding64 = pooling_descriptor.padding(); in ScopedPoolingDescriptor() local
726 std::transform(padding64.cbegin(), padding64.cend(), padding.begin(), in ScopedPoolingDescriptor()
|