Searched refs:seq_lens_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 35 const TensorShape seq_lens_shape = context->InputShape(1); in Compile() local
37 OP_REQUIRES(context, TensorShapeUtils::IsVector(seq_lens_shape), in Compile()
39 seq_lens_shape.dims())); in Compile()
52 seq_lens_shape.num_elements() == input_shape.dim_size(batch_dim_), in Compile()
54 "), ", "(", seq_lens_shape.num_elements(), in Compile()
71 OP_REQUIRES_OK(context, TensorShapeToXLAShape(seq_lens_type, seq_lens_shape, in Compile()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1248 ShapeHandle seq_lens_shape; in __anon0d47aeef2302() local
1249 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &seq_lens_shape)); in __anon0d47aeef2302()
1273 c->Merge(batch_dim_dim, c->Dim(seq_lens_shape, 0), &batch_dim_dim)); in __anon0d47aeef2302()
|