| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_sequence_op.cc | 46 const Tensor& seq_lens = context->input(1); in CheckErrors() local
48 auto seq_lens_t = seq_lens.vec<Tlen>(); in CheckErrors()
64 OP_REQUIRES(context, seq_lens.NumElements() == input.dim_size(batch_dim), in CheckErrors()
66 "), ", "(", seq_lens.NumElements(), in CheckErrors()
80 const Tensor& seq_lens = context->input(1); in CheckErrorsGPU() local
91 OP_REQUIRES(context, seq_lens.NumElements() == input.dim_size(batch_dim), in CheckErrorsGPU()
93 "), ", "(", seq_lens.NumElements(), in CheckErrorsGPU()
120 const Tensor& seq_lens = context->input(1); in Compute() local
123 OP_REQUIRES(context, TensorShapeUtils::IsVector(seq_lens.shape()), in Compute()
125 seq_lens.dims())); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | ctc_loss_op_test.py | 59 seq_lens, argument
70 inputs=inputs_t, labels=labels, sequence_length=seq_lens)
203 seq_lens = np.array([5, 5], dtype=np.int32)
217 self._testCTCLoss(inputs, seq_lens, labels, loss_truth, grad_truth)
229 seq_lens = np.array([2, 2], dtype=np.int32)
238 inputs=inputs_t, labels=labels, sequence_length=seq_lens)
242 sequence_length=seq_lens,
252 seq_lens = np.array([2, 2], dtype=np.int32)
257 inputs=inputs_t, labels=labels, sequence_length=seq_lens)
|
| D | ctc_decoder_ops_test.py | 50 seq_lens, argument
62 inputs_t, sequence_length=seq_lens, **decoder_args)
136 seq_lens = np.array([seq_len_0, seq_len_1], dtype=np.int32)
170 self._testCTCDecoder(ctc_ops.ctc_greedy_decoder, inputs, seq_lens,
201 seq_lens = np.array([seq_len_0], dtype=np.int32)
229 seq_lens,
242 seq_lens,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 59 const auto seq_lens = context->Input(1); in Compile() local
101 auto seq_lens = body_builder->GetTupleElement(param, 1); in Compile() local
106 seq_lens, body_builder->Reshape(i, {1}), {1}); in Compile()
146 seq_lens, output}); in Compile()
153 builder->Tuple({XlaHelpers::Zero(builder, seq_lens_type), seq_lens, in Compile()
163 auto mask = builder->Lt(iota, builder->Reshape(seq_lens, dims), {seq_dim_}); in Compile()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | batch_dataset_op_test.py | 214 seq_lens = array_ops.placeholder(dtypes.int32, shape=[None])
218 dataset_ops.Dataset.from_tensor_slices(seq_lens)
231 seq_lens: random_seq_lens
248 seq_lens: random_seq_lens
261 sess.run(init_op, feed_dict={padded_shape: [-1], seq_lens: [0, 0, 0, 0]})
269 sess.run(init_op, feed_dict={padded_shape: [5], seq_lens: [6, 5, 5, 5]})
274 seq_lens = array_ops.placeholder(dtypes.int32, shape=[None])
282 dataset_ops.Dataset.from_tensor_slices(seq_lens).map(fill_tuple)
297 seq_lens: random_seq_lens
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ReverseSequence.pbtxt | 62 # while entries past seq_lens are copied through:
84 # while entries past seq_lens are copied through:
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | tensor_queue_dataset_test.py | 322 def build_dataset(seq_lens): argument
323 return dataset_ops.Dataset.from_tensor_slices(seq_lens).map(
334 def build_dataset(seq_lens): argument
341 return dataset_ops.Dataset.from_tensor_slices(seq_lens).map(
|
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| D | batch_dataset_op_test.py | 511 def build_dataset(seq_lens): argument
512 return dataset_ops.Dataset.from_tensor_slices(seq_lens).map(
523 def build_dataset(seq_lens): argument
530 return dataset_ops.Dataset.from_tensor_slices(seq_lens).map(
|