Searched refs:shape0 (Results 1 - 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/contrib/graph_editor/
H A Dreroute.py63 shape0, shape1 = t0.get_shape(), t1.get_shape()
64 if not shape0.is_compatible_with(shape1):
65 raise ValueError("Shapes {} and {} are not compatible.".format(shape0,
/external/tensorflow/tensorflow/core/ops/
H A Dio_ops.cc94 ShapeHandle shape0, shape1, shape2;
95 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &shape0));
98 TF_RETURN_IF_ERROR(c->Merge(shape1, shape2, &shape0));
/external/tensorflow/tensorflow/compiler/tests/
H A Dconcat_ops_test.py259 for shape0 in (), (2,):
260 axis = len(shape0)
264 x0 = np.random.randn(*(shape0 + (n0,) + shape1))
265 x1 = np.random.randn(*(shape0 + (n1,) + shape1))
/external/tensorflow/tensorflow/contrib/lite/toco/
H A Dtooling_util.h107 // arrays d0 and d1. Without loss of generality, assume that shape0 may have
108 // higher dimensionality (length(d0) >= length(d1)). Then shape0 and shape1
113 bool ShapesAgreeUpToBroadcasting(const Shape& shape0, const Shape& shape1);
118 // arrays d0 and d1. Without loss of generality, assume that shape0 may have
119 // higher dimensionality (length(d0) >= length(d1)). Then shape0 and shape1
124 bool ShapesAgreeUpToExtending(const Shape& shape0, const Shape& shape1);
H A Dtooling_util.cc553 bool ShapesAgreeUpToBroadcasting(const Shape& shape0, const Shape& shape1) { argument
554 CheckShapeDimensions(shape0);
557 const Shape* longer = &shape0;
559 if (shape1.dimensions_count() > shape0.dimensions_count()) {
561 shorter = &shape0;
581 bool ShapesAgreeUpToExtending(const Shape& shape0, const Shape& shape1) { argument
582 CheckShapeDimensions(shape0);
585 const Shape* longer = &shape0;
587 if (shape1.dimensions_count() > shape0.dimensions_count()) {
589 shorter = &shape0;
[all...]
/external/tensorflow/tensorflow/python/kernel_tests/
H A Dconcat_op_test.py362 for shape0 in (), (2,):
363 axis = len(shape0)
367 x0 = np.random.randn(*(shape0 + (n0,) + shape1))
368 x1 = np.random.randn(*(shape0 + (n1,) + shape1))
/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util_test.cc550 Shape shape0 = ShapeUtil::MakeShape(S32, {9, 1, 4}); local
554 ShapeUtil::InsertedOrDeleted1SizedDimensions(shape0, shape1)));
556 ShapeUtil::InsertedOrDeleted1SizedDimensions(shape0, shape2)));
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dfusion_test.cc785 Shape shape0 = ShapeUtil::MakeShape(F32, {param0_dim0, param0_dim1}); local
786 auto param0 = builder.Parameter(0, shape0, "param0");

Completed in 234 milliseconds