Searched refs:packed_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn.py | 246 packed_shape = array_ops.stack(shape)
248 math_ops.reduce_all(math_ops.equal(x_shape, packed_shape)), [
249 "Expected shape for Tensor %s is " % x.name, packed_shape,
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_test.cc | 182 TensorShape packed_shape; in TEST_F() local
183 TF_ASSERT_OK(packed_handle->Shape(&packed_shape)); in TEST_F()
184 EXPECT_EQ(packed_shape, shape); in TEST_F()
235 TensorShape packed_shape; in TEST_F() local
236 TF_ASSERT_OK(packed_handle->Shape(&packed_shape)); in TEST_F()
237 EXPECT_EQ(packed_shape, shape); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1611 std::unique_ptr<Shape> packed_shape; in ProcessPackOperator() local
1620 if (!packed_shape) { in ProcessPackOperator()
1621 packed_shape.reset(new Shape(shape)); in ProcessPackOperator()
1623 CHECK(*packed_shape == shape) << "All input arrays to Pack operators " in ProcessPackOperator()
1632 axis += packed_shape->dims().size() + 1; in ProcessPackOperator()
1634 packed_shape->mutable_dims()->insert( in ProcessPackOperator()
1635 packed_shape->mutable_dims()->begin() + axis, op->inputs.size()); in ProcessPackOperator()
1636 output_array.copy_shape(*packed_shape); in ProcessPackOperator()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 668 packed_shape = array_ops.stack(shape)
670 math_ops.reduce_all(math_ops.equal(x_shape, packed_shape)), [
671 "Expected shape for Tensor %s is " % x.name, packed_shape,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 284 SmallVector<int64_t, 4> packed_shape; in matchAndRewrite() local
285 packed_shape.push_back(-1); in matchAndRewrite()
286 packed_shape.append(item_shape.begin(), item_shape.end()); in matchAndRewrite()
289 loc, GetI64ElementsAttr(packed_shape, &rewriter)); in matchAndRewrite()
|