| /external/tensorflow/tensorflow/core/kernels/ |
| reverse_sequence_op.h | 36 seq_dim_(seq_dim),
42 if (coords[seq_dim_] < seq_lengths_(coords[batch_dim_])) {
43 new_coords[seq_dim_] =
44 seq_lengths_(coords[batch_dim_]) - coords[seq_dim_] - 1;
53 int32 seq_dim_; member in class:tensorflow::generator::ReverseGenerator
|
| reverse_sequence_op.cc | 115 OP_REQUIRES_OK(context, context->GetAttr("seq_dim", &seq_dim_));
129 CheckErrors<Device, Tlen>(context, batch_dim_, seq_dim_);
141 seq_dim_, seq_lens_t, output->tensor<T, NDIM>()); \
160 int32 seq_dim_; member in class:tensorflow::ReverseSequenceOp
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| reverse_sequence_op.cc | 30 OP_REQUIRES_OK(context, context->GetAttr("seq_dim", &seq_dim_));
40 OP_REQUIRES(context, batch_dim_ != seq_dim_,
41 errors::InvalidArgument("batch_dim == seq_dim == ", seq_dim_));
43 context, seq_dim_ < input_shape.dims(),
45 seq_dim_, " vs. ", input_shape.dims(), ")"));
65 const int64 max_seq_len = input_shape.dim_size(seq_dim_);
128 seq_dim_),
154 builder->Rev(input, {seq_dim_})}));
163 auto mask = builder->Lt(iota, builder->Reshape(seq_lens, dims), {seq_dim_});
176 int32 seq_dim_; member in class:tensorflow::__anon38769::ReverseSequenceOp
[all...] |
