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.cc74 std::vector<int64> filter_layout; in StreamExecutorConvLayoutsToXlaLayouts() local
152 FilterLayout filter_layout; in XlaConvLayoutsToStreamExecutorLayouts() local
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto87 FilterLayout filter_layout = 4; field
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc244 se::dnn::FilterLayout filter_layout; in operator ()() local
Dconv_ops_3d.cc389 se::dnn::FilterLayout filter_layout; in launch() local
Dconv_grad_ops_3d.cc1293 se::dnn::FilterLayout filter_layout; in Compute() local
1787 se::dnn::FilterLayout filter_layout; in Compute() local
Dconv_grad_filter_ops.cc849 se::dnn::FilterLayout filter_layout; in operator ()() local
Dconv_ops.cc874 se::dnn::FilterLayout filter_layout; in operator ()() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc924 dnn::FilterLayout filter_layout) { in ToCudnnDataType()