HomeSort by relevance Sort by last modified time
    Searched refs:MakeShapeWithLayout (Results 1 - 25 of 34) sorted by null

1 2

  /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_test.cc 39 ShapeUtil::MakeShapeWithLayout(F32, {}, {}, {Tile({256})});
41 const Shape matrix2_ = ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1});
94 EXPECT_NE(ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0}),
95 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {0, 1}));
98 EXPECT_NE(ShapeUtil::MakeShapeWithLayout(F32, {44, 23}, {1, 0}),
99 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0}));
102 EXPECT_NE(ShapeUtil::MakeShapeWithLayout(S32, {44, 23}, {1, 0}),
103 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0}));
106 EXPECT_EQ(ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0}),
107 ShapeUtil::MakeShapeWithLayout(F32, {23, 44}, {1, 0}))
    [all...]
shape_util_test.cc 113 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1})}),
116 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1})})));
120 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1})}),
123 ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {1, 0})})));
163 ShapeUtil::MakeShapeWithLayout(F32, {4, 3}, {0, 1}),
164 ShapeUtil::MakeShapeWithLayout(F16, {4, 3}, {0, 1})));
169 ShapeUtil::MakeShapeWithLayout(F32, {4, 3}, {0, 1}),
170 ShapeUtil::MakeShapeWithLayout(F16, {3, 4}, {0, 1})));
172 ShapeUtil::MakeShapeWithLayout(F32, {3, 4}, {0, 1}),
173 ShapeUtil::MakeShapeWithLayout(F16, {3, 4}, {1, 0})))
    [all...]
literal_util.h 334 Literal literal(ShapeUtil::MakeShapeWithLayout(
438 Literal literal(ShapeUtil::MakeShapeWithLayout(
shape_util.cc 189 /* static */ Shape ShapeUtil::MakeShapeWithLayout(
202 return MakeShapeWithLayout(element_type, dimensions, layout);
    [all...]
literal_test.cc 423 Literal colmajor(ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1}));
429 Literal rowmajor(ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0}));
973 Shape shape = ShapeUtil::MakeShapeWithLayout
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
shape_partition_test.cc 47 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {1, 3}, {1, 0}), 2,
55 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {3, 1}, {0, 1}), 2,
63 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {5, 3}, {1, 0}), 6,
70 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {5, 3}, {0, 1}), 4,
80 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {5, 3, 2}, {2, 1, 0}), 16,
90 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {5, 3, 2}, {2, 0, 1}), 16,
100 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {5, 3}, {1, 0});
125 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {5, 3, 2}, {2, 1, 0});
165 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, RandR4Dims(), {3, 2, 1, 0});
cpu_layout_assignment_test.cc 66 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1});
68 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1});
101 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1});
103 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1});
143 Shape lhs_a_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1});
144 Shape lhs_b_shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 12}, {0, 1});
145 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1});
146 Shape result_a_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1});
147 Shape result_b_shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 24}, {0, 1});
185 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1})
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
client_test.cc 52 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2},
85 {ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2},
87 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2},
104 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2},
108 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2},
copy_test.cc 245 Shape in_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {0, 1});
246 Shape out_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {1, 0});
local_client_execute_test.cc 172 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, {0, 1})),
184 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, {1, 0})),
262 {ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2},
264 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2},
590 ShapeUtil::MakeShapeWithLayout(F32,
    [all...]
multioutput_fusion_test.cc 67 const Shape elem_shape0 = ShapeUtil::MakeShapeWithLayout(F32, {}, {});
69 ShapeUtil::MakeShapeWithLayout(F32, {size, size}, {1, 0});
custom_call_test.cc 174 ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0});
transfer_manager_test.cc 158 ShapeUtil::MakeShapeWithLayout(F32, {2, 3}, {1, 0});
broadcast_simple_test.cc 61 *r3_shape = ShapeUtil::MakeShapeWithLayout(F32, bounds, minor_to_major);
75 *r2_shape = ShapeUtil::MakeShapeWithLayout(F32, bounds, minor_to_major);
reshape_test.cc 620 ShapeUtil::MakeShapeWithLayout(use_bfloat16() ? BF16 : F32, {2, 8},
    [all...]
dot_operation_test.cc 321 ShapeUtil::MakeShapeWithLayout(
326 ShapeUtil::MakeShapeWithLayout(
335 ShapeUtil::MakeShapeWithLayout(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
gpu_ftz_test.cc 34 Shape param_shape = ShapeUtil::MakeShapeWithLayout(
51 Shape param_shape = ShapeUtil::MakeShapeWithLayout(
  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 311 ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0});
313 ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1});
586 ShapeUtil::MakeShapeWithLayout(F32, {3, 5, 6, 7}, {2, 0, 3, 1});
    [all...]
hlo_matchers_test.cc 109 0, ShapeUtil::MakeShapeWithLayout(F32, {5, 7}, {0, 1}), "param");
125 op::Shape(ShapeUtil::MakeShapeWithLayout(F32, {5, 7}, {0, 1})));
127 EXPECT_THAT(p0.get(), op::ShapeWithLayout(ShapeUtil::MakeShapeWithLayout(
132 ShapeUtil::MakeShapeWithLayout(F32, {5, 7}, {1, 0}))));
139 Explain(p0.get(), op::ShapeWithLayout(ShapeUtil::MakeShapeWithLayout(
pattern_matcher_test.cc 569 auto shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 2}, {0, 1});
575 ShapeUtil::MakeShapeWithLayout(F32, {1, 2}, {1, 0}),
626 ShapeUtil::MakeShapeWithLayout(F32, {1, 2}, {1, 0}),
    [all...]
algebraic_simplifier_test.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc 362 ShapeUtil::MakeShapeWithLayout(F32, {8, 8, 256, 64}, {3, 2, 1, 0});
399 Shape expected_shape = ShapeUtil::MakeShapeWithLayout(F32, {3, 2}, {1, 0});
  /external/tensorflow/tensorflow/compiler/tf2xla/
shape_util.cc 121 return xla::ShapeUtil::MakeShapeWithLayout(type, dimensions, layout);
  /external/tensorflow/tensorflow/compiler/xrt/tests/
raw_api_test.cc 174 xla::ShapeUtil::MakeShapeWithLayout(xla::F32, {2, 2}, {0, 1}), "P0");
177 xla::ShapeUtil::MakeShapeWithLayout(xla::F32, {2, 1}, {0, 1}), "P1");
    [all...]

Completed in 1382 milliseconds

1 2