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

  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 444 Shape input_shape_with_layout(ashape);
446 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({1, 0});
452 ShapeLayout(input_shape_with_layout);
585 Shape input_shape_with_layout = local
588 HloInstruction::CreateParameter(0, input_shape_with_layout, "param"));
622 Shape input_shape_with_layout(input_shape);
623 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({2, 1, 0});
625 HloInstruction::CreateParameter(0, input_shape_with_layout, "param"));
639 Shape input_shape_with_layout(input_shape);
640 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({2, 1, 0})
    [all...]
service.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
xla.proto 397 repeated ShapeProto input_shape_with_layout = 3;

Completed in 94 milliseconds