Searched defs:input_layout (Results 1 – 4 of 4) sorted by relevance
41 std::vector<int64> input_layout; in StreamExecutorConvLayoutsToXlaLayouts() local130 DataLayout input_layout; in XlaConvLayoutsToStreamExecutorLayouts() local
296 ChannelLayout input_layout, in Initialize()560 ChannelLayout input_layout, in ProcessStream()
463 std::array<int64, 4> input_layout; // minor-to-major member
643 Layout input_layout = input_is_row_major ? LayoutUtil::MakeLayout({1, 0}) in TEST() local