HomeSort by relevance Sort by last modified time
    Searched defs:MakeShapeWithLayout (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
layout_util_test.cc 29 Shape MakeShapeWithLayout(PrimitiveType element_type,
48 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 3}, {0, 1})});
50 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 2}, {0, 1})});
73 Shape src = MakeShapeWithLayout(F32, {2, 3}, {0, 1});
74 Shape dst = MakeShapeWithLayout(F32, {2, 3}, {1, 0});
97 Shape dst = MakeShapeWithLayout(F32, {2, 3}, {1, 0});
120 {MakeShapeWithLayout(F32, {2, 3}, {0, 1}),
121 MakeShapeWithLayout(F32, {42, 123}, {1, 0}),
123 {MakeShapeWithLayout(F32, {}, {}),
124 MakeShapeWithLayout(F32, {1, 2, 3}, {0, 2, 1})})})
    [all...]
shape_util.cc 191 /* static */ Shape ShapeUtil::MakeShapeWithLayout(
202 return MakeShapeWithLayout(element_type, dimensions, layout);
886 shape.has_layout() ? MakeShapeWithLayout(shape.element_type(),
    [all...]

Completed in 1829 milliseconds