Home
last modified time | relevance | path

Searched refs:packed_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn.py243 packed_shape = array_ops.stack(shape)
245 math_ops.reduce_all(math_ops.equal(x_shape, packed_shape)), [
246 "Expected shape for Tensor %s is " % x.name, packed_shape,
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1591 std::unique_ptr<Shape> packed_shape; in ProcessPackOperator() local
1600 if (!packed_shape) { in ProcessPackOperator()
1601 packed_shape.reset(new Shape(shape)); in ProcessPackOperator()
1603 CHECK(*packed_shape == shape) << "All input arrays to Pack operators " in ProcessPackOperator()
1612 axis += packed_shape->dims().size() + 1; in ProcessPackOperator()
1614 packed_shape->mutable_dims()->insert( in ProcessPackOperator()
1615 packed_shape->mutable_dims()->begin() + axis, op->inputs.size()); in ProcessPackOperator()
1616 output_array.copy_shape(*packed_shape); in ProcessPackOperator()
/external/tensorflow/tensorflow/python/ops/
Drnn.py649 packed_shape = array_ops.stack(shape)
651 math_ops.reduce_all(math_ops.equal(x_shape, packed_shape)),
653 packed_shape, " but saw shape: ", x_shape])