/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...] |