Searched refs:filter_layout (Results 1 – 6 of 6) sorted by relevance
63 std::vector<int64> filter_layout; in StreamExecutorConvLayoutsToXlaLayouts() local66 filter_layout.push_back(dnums.kernel_output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()67 filter_layout.push_back(dnums.kernel_input_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()68 filter_layout.insert(filter_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()73 filter_layout.push_back(dnums.kernel_output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()74 filter_layout.insert(filter_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()77 filter_layout.push_back(dnums.kernel_input_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()108 LayoutUtil::MakeLayoutFromMajorToMinor(filter_layout), in StreamExecutorConvLayoutsToXlaLayouts()141 FilterLayout filter_layout; in XlaConvLayoutsToStreamExecutorLayouts() local143 filter_layout = FilterLayout::kOutputInputYX; in XlaConvLayoutsToStreamExecutorLayouts()[all …]
82 FilterLayout filter_layout = 4; field
451 FilterLayout layout() const { return tensor_.filter_layout(); }
576 constexpr auto filter_layout = is_int8x4 ? dnn::FilterLayout::kOutputInputYX4 in launch() local592 .set_layout(filter_layout); in launch()
865 dnn::FilterLayout filter_layout) { in ToCudnnDataType() argument867 filter_layout == dnn::FilterLayout::kOutputInputYX4) { in ToCudnnDataType()
1548 dnn::FilterLayout filter_layout) { in ToMIOpenDataType() argument