Searched refs:shape_4x4 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes_test.py | 116 shape_4x4 = tensor_shape.matrix(4, 4)
117 self._assert_broadcast(expected=shape_4x4, shape1=shape_4, shape2=shape_4x1)
121 expected=shape_4x4, shape1=shape_1x4, shape2=shape_4x1)
203 shape_4x4 = tensor_shape.matrix(4, 4)
205 expected=shape_4x4, shape1=shape_4xU, shape2=shape_Ux4)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 1582 Shape shape_4x4 = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local
1591 HloInstruction::CreateParameter(2, shape_4x4, "param_c")); in TEST_F()
|