Home
last modified time | relevance | path

Searched defs:strides64 (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc503 std::vector<int64> strides64 = in CudnnTensorDescriptor() local
653 absl::Span<const int64> strides64 = convolution_descriptor.strides(); in CudnnConvolutionDescriptor() local
721 absl::Span<const int64> strides64 = pooling_descriptor.strides(); in CudnnPoolingDescriptor() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc546 std::vector<int64> strides64 = in ScopedTensorDescriptor() local
655 const auto& strides64 = convolution_descriptor.strides(); in ScopedConvolutionDescriptor() local
716 absl::Span<const int64> strides64 = pooling_descriptor.strides(); in ScopedPoolingDescriptor() local