| /external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
| single_image_random_dot_stereograms_ops_test.py | 60 shape_2 = sirds_2.get_shape().as_list()
61 self.assertEqual(shape_2, [768, 1024, 3])
64 self.assertAllEqual(shape_2, r_tf_2.shape)
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| linear_operator_composition_test.py | 165 shape_2 = batch_shape + [k, shape[-1]]
170 shape_2, dtype=dtype)
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util_test.cc | 163 Shape shape_2 = ShapeUtil::MakeShape(F32, {3, 2}); local
164 auto layout_2 = shape_2.mutable_layout();
169 EXPECT_FALSE(ShapeUtil::Equal(shape_1, shape_2));
170 EXPECT_TRUE(ShapeUtil::Compatible(shape_1, shape_2));
187 Shape shape_2 = ShapeUtil::MakeShape(PRED, {3, 2}); local
188 EXPECT_FALSE(ShapeUtil::Compatible(shape_1, shape_2));
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| hexagon_control_wrapper.cc | 254 const int64 shape_2 = params.shape(2); local
266 shape_2, shape_3, data_ptr, data_size);
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| op_level_cost_estimator.cc | 128 auto shape_2 = local
130 if (shape_1.dim_size() == shape_2.dim_size()) {
133 std::max(shape_1.dim(i).size(), shape_2.dim(i).size()));
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_test.cc | 1968 const int64_t shape_2[] = {2, 4, 6}; local
[all...] |
