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,
804 full_strides = [1] + strides + [1]
807 full_strides = test_util.NHWCToNCHW(full_strides)
812 strides=full_strides,
Dconv_ops_test.py280 full_strides = [1, 1] + strides
283 full_strides = [1] + strides + [1]
295 strides=full_strides,
1299 full_strides = [1] + strides + [1]
1302 full_strides = test_util.NHWCToNCHW(full_strides)
1307 strides=full_strides,
1345 full_strides = [1] + strides + [1]
1348 full_strides = test_util.NHWCToNCHW(full_strides)
1353 strides=full_strides,
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc288 std::vector<int64> BatchDescriptor::full_strides( in full_strides() function in stream_executor::dnn::BatchDescriptor
Ddnn.h268 std::vector<int64> full_strides(const DataLayout& layout) const;
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc526 batch_descriptor.full_strides(dnn::DataLayout::kBatchDepthYX); in CudnnTensorDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc663 batch_descriptor.full_strides(dnn::DataLayout::kBatchDepthYX); in ScopedTensorDescriptor()