| /external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
| single_image_random_dot_stereograms_ops_test.py | 48 shape_1 = sirds_1.get_shape().as_list()
49 self.assertEqual(shape_1, [768, 1024, 1])
52 self.assertAllEqual(shape_1, r_tf_1.shape)
|
| /external/tensorflow/tensorflow/python/framework/ |
| common_shapes_test.py | 94 shape_1 = tensor_shape.vector(1)
103 shape_0, shape_1, shape_4, shape_1x4, shape_4x1, shape_3x4, shape_4x3):
107 for identity in (shape_0, shape_1):
159 shape_1 = tensor_shape.vector(1)
174 for identity in (shape_0, shape_1):
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| linear_operator_composition_test.py | 164 shape_1 = batch_shape + [shape[-2], k]
169 shape_1, dtype=dtype), linear_operator_test_util.random_normal(
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util_test.cc | 157 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); local
158 auto layout_1 = shape_1.mutable_layout();
169 EXPECT_FALSE(ShapeUtil::Equal(shape_1, shape_2));
170 EXPECT_TRUE(ShapeUtil::Compatible(shape_1, shape_2));
186 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); local
188 EXPECT_FALSE(ShapeUtil::Compatible(shape_1, shape_2));
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| hexagon_control_wrapper.cc | 253 const int64 shape_1 = params.shape(1); local
265 node_id + NODE_ID_OFFSET, shape_0, shape_1,
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| op_level_cost_estimator.cc | 126 auto shape_1 = local
130 if (shape_1.dim_size() == shape_2.dim_size()) {
131 for (int i = 0; i < shape_1.dim_size(); i++) {
133 std::max(shape_1.dim(i).size(), shape_2.dim(i).size()));
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| nn_grad.py | 504 shape_0, shape_1 = array_ops.shape_n([op.inputs[0], op.inputs[1]])
517 shape_1,
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_test.cc | 1967 const int64_t shape_1[] = {1, 3}; local
[all...] |
