Home
last modified time | relevance | path

Searched refs:shape_a (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dmath_test.py235 shape_a = (5, 3) if tr_a else (3, 5)
237 shape_a = (2,) + shape_a
242 x = random_ops.random_uniform(shape_a)
260 shape_a = (4, 5, 3) if tr_a else (4, 3, 5)
262 shape_a = (2,) + shape_a
267 x = random_ops.random_uniform(shape_a)
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass_test.cc303 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
307 Output shape = ops::Add(s.WithOpName("shape"), shape_a, shape_b); in TEST()
355 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
359 Output shape = ops::Add(s.WithOpName("shape"), shape_a, shape_b); in TEST()
380 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), 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/
Dgraph_properties_test.cc1566 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
1568 EXPECT_EQ(2, shape_a.dim_size()); in TEST_F()
1569 EXPECT_EQ(shape_a.dim_size(), shape_c.dim_size()); in TEST_F()
1570 EXPECT_GE(-2, shape_a.dim(0).size()); in TEST_F()
1571 EXPECT_EQ(shape_a.dim(0).size(), shape_c.dim(0).size()); in TEST_F()
1572 EXPECT_GE(-2, shape_a.dim(1).size()); in TEST_F()
1573 EXPECT_EQ(shape_a.dim(1).size(), shape_c.dim(1).size()); in TEST_F()
1575 PartialTensorShape shape(shape_a); in TEST_F()
1584 EXPECT_NE(shape_a.dim(0).size(), shape_b.dim(0).size()); in TEST_F()
1595 EXPECT_EQ(shape_f.dim(0).size(), shape_a.dim(0).size()); in TEST_F()
[all …]
/external/tensorflow/tensorflow/python/ops/
Dmath_ops.py3626 shape_a = a.get_shape().as_list()
3627 axes = [i if i >= 0 else i + len(shape_a) for i in axes]
3628 free = [i for i in xrange(len(shape_a)) if i not in axes]
3629 free_dims = [shape_a[i] for i in free]
3630 prod_free = int(np.prod([shape_a[i] for i in free]))
3631 prod_axes = int(np.prod([shape_a[i] for i in axes]))
3638 shape_a = a.get_shape().as_list()
3639 axes = [i if i >= 0 else i + len(shape_a) for i in axes]
3640 free = [i for i in xrange(len(shape_a)) if i not in axes]
3641 axes_dims = [shape_a[i] for i in axes]
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc189 Shape shape_a = ShapeUtil::MakeShape(F32, {4, 3}, {true, false}); in TEST() local
190 *shape_a.mutable_layout() = Layout({1, 0}); in TEST()
196 EXPECT_TRUE(ShapeUtil::Compatible(shape_a, shape_a)); in TEST()
197 EXPECT_TRUE(ShapeUtil::Compatible(shape_a, shape_b)); in TEST()
198 EXPECT_FALSE(ShapeUtil::Compatible(shape_a, shape_c)); in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test.cc147 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_a = in BuildRemoteFusedGraphExecuteInfo() local
149 shape_a.set_dtype(DT_FLOAT); in BuildRemoteFusedGraphExecuteInfo()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h88 const RuntimeShape* shape_a = in ProcessBroadcastShapes() local
101 while (i >= 0 && shape_a->Dims(i) == shape_b->Dims(i)) { in ProcessBroadcastShapes()
107 while (i >= 0 && shape_a->Dims(i) == 1) { in ProcessBroadcastShapes()
111 while (i >= 0 && shape_a->Dims(i) == shape_b->Dims(i)) { in ProcessBroadcastShapes()
112 params->broadcast_shape[2] *= shape_a->Dims(i); in ProcessBroadcastShapes()
117 params->broadcast_shape[1] *= shape_a->Dims(i); in ProcessBroadcastShapes()
120 while (i >= 0 && shape_a->Dims(i) == shape_b->Dims(i)) { in ProcessBroadcastShapes()