Searched refs:filter_layout (Results 1 – 10 of 10) sorted by relevance
74 std::vector<int64> filter_layout; in StreamExecutorConvLayoutsToXlaLayouts() local77 filter_layout.push_back(dnums.kernel_output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()78 filter_layout.push_back(dnums.kernel_input_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()79 filter_layout.insert(filter_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()84 filter_layout.push_back(dnums.kernel_output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()85 filter_layout.insert(filter_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()88 filter_layout.push_back(dnums.kernel_input_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()119 LayoutUtil::MakeLayoutFromMajorToMinor(filter_layout), in StreamExecutorConvLayoutsToXlaLayouts()152 FilterLayout filter_layout; in XlaConvLayoutsToStreamExecutorLayouts() local154 filter_layout = FilterLayout::kOutputInputYX; in XlaConvLayoutsToStreamExecutorLayouts()[all …]
30 // Note: Not using names like input_layout or filter_layout as `input` may be
87 FilterLayout filter_layout = 4; field
443 FilterLayout layout() const { return tensor_.filter_layout(); } in layout()
244 se::dnn::FilterLayout filter_layout; in operator ()() local246 std::tie(compute_data_layout, filter_layout) = in operator ()()266 .set_layout(filter_layout); in operator ()()
389 se::dnn::FilterLayout filter_layout; in launch() local391 std::tie(compute_data_layout, filter_layout) = in launch()417 .set_layout(filter_layout); in launch()
1293 se::dnn::FilterLayout filter_layout; in Compute() local1295 std::tie(compute_data_layout, filter_layout) = in Compute()1318 .set_layout(filter_layout); in Compute()1787 se::dnn::FilterLayout filter_layout; in Compute() local1789 std::tie(compute_data_layout, filter_layout) = in Compute()1815 .set_layout(filter_layout); in Compute()
849 se::dnn::FilterLayout filter_layout; in operator ()() local851 std::tie(compute_data_layout, filter_layout) = in operator ()()871 .set_layout(filter_layout); in operator ()()
874 se::dnn::FilterLayout filter_layout; in operator ()() local876 std::tie(compute_data_layout, filter_layout) = in operator ()()896 .set_layout(filter_layout); in operator ()()
924 dnn::FilterLayout filter_layout) { in ToCudnnDataType() argument926 filter_layout == dnn::FilterLayout::kOutputInputYX4) { in ToCudnnDataType()