| /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_test.cc | 113 Shape expected = ShapeUtil::MakeShapeWithLayout(F32, {123, 456}, {0, 1});
123 Shape expected = ShapeUtil::MakeShapeWithLayout(F32, {123, 456}, {0, 1});
267 ShapeUtil::MakeShapeWithLayout(F32, {}, {});
442 Shape matrix2 = ShapeUtil::MakeShapeWithLayout(S32, {3, 4}, {0, 1});
647 ShapeUtil::MakeShapeWithLayout(
649 ShapeUtil::MakeShapeWithLayout(
658 ShapeUtil::MakeShapeWithLayout(F32, {3, 2, 2}, {1, 0, 2}),
659 ShapeUtil::MakeShapeWithLayout(F32, {6, 2}, {0, 1})));
664 ShapeUtil::MakeShapeWithLayout(F32, {3, 2, 2}, {0, 1, 2}),
665 ShapeUtil::MakeShapeWithLayout(F32, {6, 2}, {0, 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...] |
| shape_util.h | 325 static Shape MakeShapeWithLayout(
|
| literal_util_test.cc | 339 MakeUnique<Literal>(ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1}));
346 MakeUnique<Literal>(ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0}));
841 Shape shape = ShapeUtil::MakeShapeWithLayout(
[all...] |
| literal_util.h | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| shape_partition_test.cc | 49 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {1, 3}, {1, 0}), 1);
53 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {3, 1}, {0, 1}), 1);
57 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {5, 3}, {1, 0}), 5);
61 RunR2Test(ShapeUtil::MakeShapeWithLayout(F32, {5, 3}, {0, 1}), 3);
65 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {5, 3, 2}, {2, 1, 0});
98 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {5, 3, 2}, {2, 0, 1});
140 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {5, 3}, {1, 0});
165 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {5, 3, 2}, {2, 1, 0});
205 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, RandR4Dims(), {3, 2, 1, 0});
|
| cpu_layout_assignment_test.cc | 59 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1});
61 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1});
94 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1});
96 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1});
136 Shape lhs_a_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1});
137 Shape lhs_b_shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 12}, {0, 1});
138 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1});
139 Shape result_a_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1});
140 Shape result_b_shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 24}, {0, 1});
178 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1})
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| client_test.cc | 51 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2},
84 {ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2},
86 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2},
102 ShapeUtil::MakeShapeWithLayout(S32, /*dimensions=*/{2, 2},
106 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 | 180 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, {0, 1})),
192 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2}, {1, 0})),
274 {ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2},
276 ShapeUtil::MakeShapeWithLayout(F32, /*dimensions=*/{2, 2},
616 ShapeUtil::MakeShapeWithLayout(F32,
[all...] |
| transfer_manager_test.cc | 147 ShapeUtil::MakeShapeWithLayout(F32, {2, 3}, {1, 0});
|
| broadcast_simple_test.cc | 61 *r3_shape = ShapeUtil::MakeShapeWithLayout(F32, bounds, minor_to_major);
74 *r2_shape = ShapeUtil::MakeShapeWithLayout(F32, bounds, minor_to_major);
|
| client_library_test_base.cc | 175 auto layout = ShapeUtil::MakeShapeWithLayout(
|
| reduce_window_test.cc | 331 ShapeUtil::MakeShapeWithLayout(F32, output_dims, output_layout);
619 const Shape& expected_shape_with_layout = ShapeUtil::MakeShapeWithLayout(
[all...] |
| reshape_test.cc | 651 ShapeUtil::MakeShapeWithLayout(use_bfloat16() ? BF16 : F32, {2, 8},
[all...] |
| fusion_test.cc | 269 ShapeUtil::MakeShapeWithLayout(F32, {rand_dim0_size, dim1_size}, {1, 0});
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| shape_util.cc | 56 *shape = xla::ShapeUtil::MakeShapeWithLayout(type, dimensions, layout);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| layout_assignment_test.cc | 282 ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0});
284 ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1});
561 ShapeUtil::MakeShapeWithLayout(F32, {3, 5, 6, 7}, {2, 0, 3, 1});
620 {ShapeUtil::MakeShapeWithLayout(F32, {2, 2, 2}, {0, 1, 2}),
622 ShapeUtil::MakeShapeWithLayout(F32, {2, 2, 2}, {1, 2, 0}),
623 ShapeUtil::MakeShapeWithLayout(F32, {2, 2, 2}, {2, 0, 1}),
694 Shape xshape = ShapeUtil::MakeShapeWithLayout(F32, {128, 8}, {0, 1});
[all...] |
| algebraic_simplifier.cc | [all...] |
| hlo_instruction_test.cc | 715 auto shape10 = ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0});
716 auto shape01 = ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1});
[all...] |
| layout_assignment.cc | [all...] |
| algebraic_simplifier_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| numpy_bridge.cc | 275 return ShapeUtil::MakeShapeWithLayout(element_type, dimensions,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| ir_emitter_unnested.cc | [all...] |
