Searched refs:seq_dim (Results 1 - 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dreverse_sequence_op.cc44 void CheckErrors(OpKernelContext* context, int batch_dim, int seq_dim) { argument
56 OP_REQUIRES(context, batch_dim != seq_dim,
57 errors::InvalidArgument("batch_dim == seq_dim == ", seq_dim));
58 OP_REQUIRES(context, seq_dim < input.dims(),
59 errors::InvalidArgument("seq_dim must be < input.dims()", "( ",
60 seq_dim, " vs. ", input.dims(), ")"));
72 OP_REQUIRES(context, seq_lens_vec[d] <= input.dim_size(seq_dim),
74 seq_dim, ")"));
78 void CheckErrorsGPU(OpKernelContext* context, int batch_dim, int seq_dim) { argument
98 CheckErrors(OpKernelContext* context, int batch_dim, int seq_dim) argument
104 CheckErrors(OpKernelContext* context, int batch_dim, int seq_dim) argument
[all...]
H A Dreverse_sequence_op.h33 int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths)
36 seq_dim_(seq_dim),
65 int32 batch_dim, int32 seq_dim,
69 seq_dim, seq_lengths);
32 ReverseGenerator(typename TTypes<T, Dims>::ConstTensor input, int32 batch_dim, int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths) argument
63 Compute( const Device& d, typename TTypes<T, Dims>::ConstTensor input, int32 batch_dim, int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths, typename TTypes<T, Dims>::Tensor output) argument
/external/tensorflow/tensorflow/python/ops/
H A Drnn.py429 def _reverse(input_, seq_lengths, seq_dim, batch_dim):
433 seq_dim=seq_dim, batch_dim=batch_dim)
435 return array_ops.reverse(input_, axis=[seq_dim])
440 seq_dim=time_dim, batch_dim=batch_dim)
449 seq_dim=time_dim, batch_dim=batch_dim)
H A Darray_ops.py2640 seq_dim=None,
2643 "seq_dim", seq_dim)
2649 seq_dim=seq_axis,
2659 "seq_dim", "seq_axis")
/external/tensorflow/tensorflow/cc/gradients/
H A Darray_grad.cc230 int seq_dim; local
231 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "seq_dim", &seq_dim));
233 ReverseSequence(scope, grad_inputs[0], seq_lengths, seq_dim,
/external/tensorflow/tensorflow/contrib/crf/python/ops/
H A Dcrf.py490 backpointers, sequence_length - 1, seq_dim=1)
508 decode_tags, sequence_length, seq_dim=1)
/external/tensorflow/tensorflow/core/ops/
H A Darray_ops_test.cc1060 auto rebuild_node_def = [&op](const int32 seq_dim, const int32 batch_dim) {
1064 .Attr("seq_dim", seq_dim)
1076 // Validate seq_dim and batch_dim
1080 INFER_ERROR("seq_dim must be < input rank", op, "[1,2,3];[3]");
H A Darray_ops.cc1242 .Attr("seq_dim: int")
1251 int64 seq_dim;
1252 TF_RETURN_IF_ERROR(c->GetAttr("seq_dim", &seq_dim));
1260 // Validate batch_dim and seq_dim against input.
1266 if (seq_dim >= input_rank) {
1268 "seq_dim must be < input rank: ", seq_dim, " vs. ", input_rank);

Completed in 351 milliseconds