Searched refs:shape0 (Results 1 - 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/contrib/graph_editor/ |
H A D | reroute.py | 63 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 D | io_ops.cc | 94 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 D | concat_ops_test.py | 259 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 D | tooling_util.h | 107 // 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 D | tooling_util.cc | 553 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 D | concat_op_test.py | 362 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 D | shape_util_test.cc | 550 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 D | fusion_test.cc | 785 Shape shape0 = ShapeUtil::MakeShape(F32, {param0_dim0, param0_dim1}); local 786 auto param0 = builder.Parameter(0, shape0, "param0");
|
Completed in 234 milliseconds