Searched refs:seq_lens_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 38 const TensorShape seq_lens_shape = context->InputShape(1); in Compile() local
40 OP_REQUIRES(context, TensorShapeUtils::IsVector(seq_lens_shape), in Compile()
42 seq_lens_shape.dims())); in Compile()
55 seq_lens_shape.num_elements() == input_shape.dim_size(batch_dim_), in Compile()
57 "), ", "(", seq_lens_shape.num_elements(), in Compile()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1477 ShapeHandle seq_lens_shape; in __anon7c94107b2402() local
1478 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &seq_lens_shape)); in __anon7c94107b2402()
1502 c->Merge(batch_dim_dim, c->Dim(seq_lens_shape, 0), &batch_dim_dim)); in __anon7c94107b2402()
|