Searched refs:shape_2 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
| D | 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/ |
| D | linear_operator_composition_test.py | 158 shape_2 = batch_shape + [k, shape[-1]]
163 shape_2, dtype=dtype)
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 133 Shape shape_2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
134 auto layout_2 = shape_2.mutable_layout(); in TEST()
139 EXPECT_FALSE(ShapeUtil::Equal(shape_1, shape_2)); in TEST()
140 EXPECT_TRUE(ShapeUtil::Compatible(shape_1, shape_2)); in TEST()
157 Shape shape_2 = ShapeUtil::MakeShape(PRED, {3, 2}); in TEST() local
158 EXPECT_FALSE(ShapeUtil::Compatible(shape_1, shape_2)); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_control_wrapper.cc | 254 const int64 shape_2 = params.shape(2); in SetupGraph() local
266 shape_2, shape_3, data_ptr, data_size); in SetupGraph()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | README.md | 183 shape_2 = TensorShape({3, 8})
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 162 auto shape_2 = in CwiseOutputElementCount() local
164 if (shape_1.dim_size() == shape_2.dim_size()) { in CwiseOutputElementCount()
167 std::max(shape_1.dim(i).size(), shape_2.dim(i).size())); in CwiseOutputElementCount()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2168 const int64_t shape_2[] = {2, 4, 6}; in TEST_F() local
2169 const int64_t* list[] = {&shape_1[0], &shape_2[0]}; in TEST_F()
2171 const int ndims[] = {TF_ARRAYSIZE(shape_1), TF_ARRAYSIZE(shape_2)}; in TEST_F()
|