Searched refs:shape_3x4 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes_test.py | 98 shape_3x4 = tensor_shape.TensorShape([3, 4])
103 shape_0, shape_1, shape_4, shape_1x4, shape_4x1, shape_3x4, shape_4x3):
108 for shape in (shape_4, shape_1x4, shape_4x1, shape_3x4, shape_4x3):
112 for shape in (shape_4, shape_1x4, shape_4x1, shape_3x4, shape_4x3):
118 self._assert_broadcast(expected=shape_3x4, shape1=shape_4, shape2=shape_3x4)
123 expected=shape_3x4, shape1=shape_1x4, shape2=shape_3x4)
125 self._assert_incompatible_broadcast(shape1=shape_4x1, shape2=shape_3x4)
128 self._assert_incompatible_broadcast(shape1=shape_3x4, shape2=shape_4x3)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 1753 Shape shape_3x4 = ShapeUtil::MakeShape(F32, {3, 4}); in TEST_F() local
1761 HloInstruction::CreateParameter(1, shape_3x4, "param_b")); in TEST_F()
1773 shape_3x4, param_b, param_c, dot_dnums, precision_config)); in TEST_F()
|