Searched defs:output_layout (Results 1 – 17 of 17) sorted by relevance
85 std::vector<int64> output_layout; in StreamExecutorConvLayoutsToXlaLayouts() local152 DataLayout output_layout; in XlaConvLayoutsToStreamExecutorLayouts() local
94 TPUEmbeddingOutputLayout output_layout = 8; field
42 std::vector<int64> output_layout; member
41 std::vector<int64> output_layout; member
43 std::vector<int64> output_layout; member
74 Layout* output_layout = nullptr) { in ComputeConstantLiteral()
369 std::vector<int64> output_layout = {1, 5, 3, 2, 0, 4}; in XLA_TEST_P() local
582 AudioProcessing::ChannelLayout output_layout = AudioProcessing::kMono; in void_main() local
297 ChannelLayout output_layout, in Initialize()562 ChannelLayout output_layout, in ProcessStream()
447 LayoutProto output_layout = 2; field
645 Layout output_layout = output_is_row_major in TEST() local
1359 std::vector<int64> output_layout; in AlignLayouts() local
1088 absl::optional<Layout> output_layout; in ComputeConstantGraph() local
988 const Layout& output_layout, const HloInstruction* instruction, in ChooseOperandLayoutFromOutputLayout()
436 const Tensor& output_layout = *GetOutput(2); in Run() local
816 dnnLayout_t output_layout = static_cast<dnnLayout_t>(mkl_shape.GetTfLayout()); in ConvertMklToTF() local
2406 const Layout& output_layout = output_shape.layout(); in EmitFastConcatenate() local