Searched refs:dims64 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 548 std::vector<int64> dims64 = in ScopedTensorDescriptor() local
556 std::transform(dims64.cbegin(), dims64.cend(), dims.begin(), in ScopedTensorDescriptor()
3604 std::vector<int64> dims64 = in DoPoolBackward() local
3610 std::transform(dims64.cbegin(), dims64.cend(), dims.begin(), in DoPoolBackward()
3700 std::vector<int64> dims64 = in DoPoolBackward() local
3706 std::transform(dims64.cbegin(), dims64.cend(), dims.begin(), in DoPoolBackward()
3835 std::vector<int64> dims64 = in DoNormalizeBackwardWithDimensions() local
3841 std::transform(dims64.cbegin(), dims64.cend(), dimsint.begin(), in DoNormalizeBackwardWithDimensions()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 505 std::vector<int64> dims64 = in CudnnTensorDescriptor() local
513 std::transform(dims64.cbegin(), dims64.cend(), dims.begin(), in CudnnTensorDescriptor()
|