Home
last modified time | relevance | path

Searched defs:filter_layout (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc102 std::vector<int64> filter_layout; in StreamExecutorConvLayoutsToXlaLayouts() local
224 FilterLayout filter_layout; in XlaConvShapesToStreamExecutorLayouts() local
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto96 FilterLayout filter_layout = 4; field
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc391 se::dnn::FilterLayout filter_layout; in launch() local
Dconv_grad_input_ops.cc235 se::dnn::FilterLayout filter_layout; in operator ()() local
Dconv_grad_ops_3d.cc1382 se::dnn::FilterLayout filter_layout; in Compute() local
1938 se::dnn::FilterLayout filter_layout; in Compute() local
Dconv_grad_filter_ops.cc855 se::dnn::FilterLayout filter_layout; in operator ()() local
Dconv_ops.cc996 se::dnn::FilterLayout filter_layout; in operator ()() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1016 dnn::FilterLayout filter_layout) { in ToCudnnDataType()
1035 cudnnDataType_t GetCudnnDataType(dnn::FilterLayout filter_layout) { in GetCudnnDataType()