Searched refs:seq_dim_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 33 OP_REQUIRES_OK(context, context->GetAttr("seq_dim", &seq_dim_)); in ReverseSequenceOp()
43 OP_REQUIRES(context, batch_dim_ != seq_dim_, in Compile()
44 errors::InvalidArgument("batch_dim == seq_dim == ", seq_dim_)); in Compile()
46 context, seq_dim_ < input_shape.dims(), in Compile()
48 seq_dim_, " vs. ", input_shape.dims(), ")")); in Compile()
71 const int64 max_seq_len = input_shape.dim_size(seq_dim_); in Compile()
87 if (batch_dim_ > seq_dim_) { in Compile()
105 dnums.add_start_index_map(seq_dim_); in Compile()
110 if (i != batch_dim_ && i != seq_dim_) { in Compile()
119 slice_sizes[seq_dim_] = 1; in Compile()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_sequence_op.h | 36 seq_dim_(seq_dim), in ReverseGenerator()
42 if (coords[seq_dim_] < seq_lengths_(coords[batch_dim_])) { in operator()
43 new_coords[seq_dim_] = in operator()
44 seq_lengths_(coords[batch_dim_]) - coords[seq_dim_] - 1; in operator()
53 int32 seq_dim_; variable
|
| D | reverse_sequence_op.cc | 115 OP_REQUIRES_OK(context, context->GetAttr("seq_dim", &seq_dim_)); in ReverseSequenceOp()
129 CheckErrors<Device, Tlen>(context, batch_dim_, seq_dim_); in Compute()
142 seq_dim_, seq_lens_t, output->tensor<T, NDIM>()); \ in Compute()
161 int32 seq_dim_; member in tensorflow::ReverseSequenceOp
|