Home
last modified time | relevance | path

Searched refs:seq_length (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_ops_benchmark.py68 seq_length = config["seq_length"]
70 return "y%d_u%d_b%d_q%d" % (num_layers, num_units, batch_size, seq_length)
94 seq_length = config["seq_length"]
100 array_ops.ones([seq_length, batch_size, num_units]))
126 seq_length = config["seq_length"]
129 inputs = array_ops.zeros([batch_size, seq_length, num_units],
150 seq_length = config["seq_length"]
153 inputs = array_ops.zeros([batch_size, seq_length, num_units],
Dcudnn_rnn_ops_test.py296 seq_length = 10
318 np.random.randn(seq_length, batch_size, input_size), dtype=dtype)
420 batch_size, seq_length, dir_count, dropout, argument
434 input_data = array_ops.ones([seq_length, batch_size, input_size])
542 batch_size, seq_length, dir_count, dropout, dtype, argument
564 [seq_length, batch_size, input_size], dtype=dtype),
598 (input_data, [seq_length, batch_size, input_size]),
Dcudnn_rnn_test.py162 def SynthesizeInput(self, seq_length, batch_size, seed=1234): argument
171 inputs = np.random.randn(seq_length, batch_size,
621 seq_length = 8
646 inputs, initial_state = model.SynthesizeInput(seq_length, batch_size)
765 def _TestCudnnCompatibleRnnCells(self, num_layers, seq_length, num_units, argument
793 inputs, _ = model.SynthesizeInput(seq_length, batch_size, seed)
822 inference_input = np.random.rand(seq_length, batch_size,
835 dtype, shape=[seq_length, batch_size, input_size])
1047 batch_size, seq_length, dir_count, dropout, dtype, argument
1076 [seq_length, batch_size, input_size], dtype=dtype),
/external/icu/icu4c/source/i18n/
Dcsr2022.cpp49 int32_t seq_length = (int32_t)uprv_strlen((const char *) seq); in match_2022() local
51 if (textLen-i >= seq_length) { in match_2022()
53 while(j < seq_length) { in match_2022()
62 i += seq_length-1; in match_2022()
/external/tensorflow/tensorflow/contrib/cudnn_rnn/ops/
Dcudnn_rnn_ops_test.cc43 int seq_length = 2; in TEST() local
48 std::vector<int> input_shape = {seq_length, batch_size, num_units}; in TEST()
51 std::vector<int> output_shape = {seq_length, batch_size, in TEST()
Dcudnn_rnn_ops.cc144 auto seq_length = c->Dim(input_shape, 0); in __anonb0631e310302() local
154 auto output_shape = c->MakeShape({seq_length, batch_size, output_size}); in __anonb0631e310302()
/external/tensorflow/tensorflow/compiler/tests/
Dlstm.py124 def RandomInputs(batch_size, seq_length, num_inputs): argument
129 for seq in range(seq_length):
137 def BuildLSTMLayer(batch_size, seq_length, num_inputs, num_nodes): argument
155 x_seq, pad_seq = RandomInputs(batch_size, seq_length, num_inputs)
Dlstm_test.py142 seq_length = 3
147 x_seq = [constant_op.constant(self._inputs)] * seq_length
162 seq_length = 3
167 x_seq = [constant_op.constant(self._inputs)] * seq_length
169 ] * seq_length
242 out_seq, weights = lstm.BuildLSTMLayer(FLAGS.batch_size, FLAGS.seq_length,
252 '%s_%d_%d_%d_%d' % (name, FLAGS.batch_size, FLAGS.seq_length,
/external/v8/src/heap/
Dmarking.h213 CellPrinter() : seq_start(0), seq_type(0), seq_length(0) {} in CellPrinter()
217 seq_length++; in Print()
225 seq_length = 0; in Print()
236 if (seq_length > 0) { in Flush()
238 seq_length * kBitsPerCell); in Flush()
239 seq_length = 0; in Flush()
248 uint32_t seq_length; variable
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1305 CudnnRnnSequenceTensorDescriptor(CUDAExecutor* parent, int seq_length, in CudnnRnnSequenceTensorDescriptor() argument
1309 seq_length_(seq_length), in CudnnRnnSequenceTensorDescriptor()
1314 if (seq_length <= 0) { in CudnnRnnSequenceTensorDescriptor()
1316 port::StrCat("sequence length must be positive: ", seq_length); in CudnnRnnSequenceTensorDescriptor()
1331 handles_.assign(seq_length, handle); in CudnnRnnSequenceTensorDescriptor()
1348 int seq_length() const { return seq_length_; } in seq_length() function in perftools::gputools::cuda::CudnnRnnSequenceTensorDescriptor
1416 int seq_length = 0; member
1440 model_dims->seq_length = input_desc.seq_length(); in ExtractAndCheckRnnForward()
1460 if (!(output_desc.seq_length() == model_dims->seq_length && in ExtractAndCheckRnnForward()
1509 input_desc.seq_length() /*seqLength*/, input_desc.handles() /*xDesc*/, in CreateRnnWorkspace()
[all …]
Dcuda_dnn.h57 createRnnSequenceTensorDescriptor(int seq_length, int batch_size,
/external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/
Dcudnn_rnn_ops.cc355 int seq_length; member
411 model_shapes->seq_length = (*input)->dim_size(0); in ExtractForwardInput()
443 TensorShape({model_shapes->seq_length, model_shapes->batch_size, in ExtractForwardInput()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc360 StreamExecutor::createRnnSequenceTensorDescriptor(int seq_length, in createRnnSequenceTensorDescriptor() argument
368 return dnn_support->createRnnSequenceTensorDescriptor(seq_length, batch_size, in createRnnSequenceTensorDescriptor()
Dstream_executor_pimpl.h381 createRnnSequenceTensorDescriptor(int seq_length, int batch_size,
Ddnn.h1993 createRnnSequenceTensorDescriptor(int seq_length, int batch_size,