Searched refs:packed_shape (Results 1 – 3 of 3) sorted by relevance
243 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,
1591 std::unique_ptr<Shape> packed_shape; in ProcessPackOperator() local1600 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()
649 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])