Searched refs:shape_b (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 238 shape_b = (7, 5) if tr_b else (5, 7)
240 shape_b = (2,) + shape_b
243 y = random_ops.random_uniform(shape_b)
263 shape_b = (4, 7, 5) if tr_b else (4, 5, 7)
265 shape_b = (2,) + shape_b
268 y = random_ops.random_uniform(shape_b)
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass_test.cc | 305 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
307 Output shape = ops::Add(s.WithOpName("shape"), shape_a, shape_b); in TEST()
357 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
359 Output shape = ops::Add(s.WithOpName("shape"), shape_a, shape_b); in TEST()
382 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
384 Output shape = ops::Add(s.WithOpName("shape"), shape_a, shape_b); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1579 const auto shape_b = properties.GetOutputProperties("b").at(0).shape(); in TEST_F() local
1581 EXPECT_EQ(1, shape_b.dim_size()); in TEST_F()
1582 EXPECT_EQ(shape_b.dim_size(), shape_d.dim_size()); in TEST_F()
1583 EXPECT_GE(-2, shape_b.dim(0).size()); in TEST_F()
1584 EXPECT_NE(shape_a.dim(0).size(), shape_b.dim(0).size()); in TEST_F()
1585 EXPECT_EQ(shape_b.dim(0).size(), shape_d.dim(0).size()); in TEST_F()
1651 const auto shape_b = properties.GetOutputProperties("b").at(0).shape(); in TEST_F() local
1655 EXPECT_EQ(shape_b.DebugString(), shape_o2.DebugString()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 191 Shape shape_b = ShapeUtil::MakeShape(F32, {4, 3}, {true, false}); in TEST() local
192 *shape_b.mutable_layout() = Layout({0, 1}); in TEST()
197 EXPECT_TRUE(ShapeUtil::Compatible(shape_a, shape_b)); in TEST()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 90 const RuntimeShape* shape_b = in ProcessBroadcastShapes() local
101 while (i >= 0 && shape_a->Dims(i) == shape_b->Dims(i)) { in ProcessBroadcastShapes()
102 params->broadcast_shape[4] *= shape_b->Dims(i); in ProcessBroadcastShapes()
108 params->broadcast_shape[3] *= shape_b->Dims(i); in ProcessBroadcastShapes()
111 while (i >= 0 && shape_a->Dims(i) == shape_b->Dims(i)) { in ProcessBroadcastShapes()
116 while (i >= 0 && shape_b->Dims(i) == 1) { in ProcessBroadcastShapes()
120 while (i >= 0 && shape_a->Dims(i) == shape_b->Dims(i)) { in ProcessBroadcastShapes()
121 params->broadcast_shape[0] *= shape_b->Dims(i); in ProcessBroadcastShapes()
|