Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc63 std::vector<int64> filter_layout; in StreamExecutorConvLayoutsToXlaLayouts() local
141 FilterLayout filter_layout; in XlaConvLayoutsToStreamExecutorLayouts() local
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto82 FilterLayout filter_layout = 4; field
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc576 constexpr auto filter_layout = is_int8x4 ? dnn::FilterLayout::kOutputInputYX4 in launch() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc865 dnn::FilterLayout filter_layout) { in ToCudnnDataType()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1548 dnn::FilterLayout filter_layout) { in ToMIOpenDataType()