Home
last modified time | relevance | path

Searched refs:full_strides (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_3d_test.py145 full_strides = [1, 1] + strides
148 full_strides = [1] + strides + [1]
160 strides=full_strides,
708 full_strides = [1] + strides + [1]
711 full_strides = test_util.NHWCToNCHW(full_strides)
716 strides=full_strides,
Dconv_ops_test.py275 full_strides = [1, 1] + strides
278 full_strides = [1] + strides + [1]
290 strides=full_strides,
1086 full_strides = [1] + strides + [1]
1089 full_strides = test_util.NHWCToNCHW(full_strides)
1094 strides=full_strides,
1132 full_strides = [1] + strides + [1]
1135 full_strides = test_util.NHWCToNCHW(full_strides)
1140 strides=full_strides,
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc257 std::vector<int64> BatchDescriptor::full_strides( in full_strides() function in stream_executor::dnn::BatchDescriptor
Ddnn.h276 std::vector<int64> full_strides(const DataLayout& layout) const;
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc504 batch_descriptor.full_strides(dnn::DataLayout::kBatchDepthYX); in CudnnTensorDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc547 batch_descriptor.full_strides(dnn::DataLayout::kBatchDepthYX); in ScopedTensorDescriptor()