HomeSort by relevance Sort by last modified time
    Searched refs:shape0 (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/contrib/graph_editor/
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/
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/
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/
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);
tooling_util.cc 553 bool ShapesAgreeUpToBroadcasting(const Shape& shape0, const Shape& shape1) {
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) {
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/
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/
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/
fusion_test.cc 785 Shape shape0 = ShapeUtil::MakeShape(F32, {param0_dim0, param0_dim1}); local
    [all...]

Completed in 1058 milliseconds