HomeSort by relevance Sort by last modified time
    Searched full:input_shape_with_layout (Results 1 - 1 of 1) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 416 Shape input_shape_with_layout(ashape);
418 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({1, 0});
424 ShapeLayout(input_shape_with_layout);
560 Shape input_shape_with_layout = local
563 HloInstruction::CreateParameter(0, input_shape_with_layout, "param"));
    [all...]

Completed in 83 milliseconds