Searched refs:strides64 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 503 std::vector<int64> strides64 = in CudnnTensorDescriptor() local
511 std::transform(strides64.cbegin(), strides64.cend(), strides.begin(), in CudnnTensorDescriptor()
653 absl::Span<const int64> strides64 = convolution_descriptor.strides(); in CudnnConvolutionDescriptor() local
664 std::transform(strides64.cbegin(), strides64.cend(), strides.begin(), in CudnnConvolutionDescriptor()
721 absl::Span<const int64> strides64 = pooling_descriptor.strides(); in CudnnPoolingDescriptor() local
729 std::transform(strides64.cbegin(), strides64.cend(), strides.begin(), in CudnnPoolingDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 546 std::vector<int64> strides64 = in ScopedTensorDescriptor() local
554 std::transform(strides64.cbegin(), strides64.cend(), strides.begin(), in ScopedTensorDescriptor()
655 const auto& strides64 = convolution_descriptor.strides(); in ScopedConvolutionDescriptor() local
665 std::transform(strides64.cbegin(), strides64.cend(), strides.begin(), in ScopedConvolutionDescriptor()
716 absl::Span<const int64> strides64 = pooling_descriptor.strides(); in ScopedPoolingDescriptor() local
724 std::transform(strides64.cbegin(), strides64.cend(), strides.begin(), in ScopedPoolingDescriptor()
|