Home
last modified time | relevance | path

Searched refs:seq_array (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tfe_src.cc1402 PyObject** seq_array = PySequence_Fast_ITEMS(seq); in CallBackwardFunction() local
1405 PyObject* item = seq_array[i]; in CallBackwardFunction()
1976 PyObject** seq_array = PySequence_Fast_ITEMS(seq); local
1980 PyObject* item = seq_array[i];
2006 PyObject** seq_array = PySequence_Fast_ITEMS(seq.get()); local
2010 PyObject* item = seq_array[i];
2142 PyObject** seq_array = PySequence_Fast_ITEMS(seq.get()); local
2144 PyObject* item = seq_array[i];
2252 PyObject** seq_array = PySequence_Fast_ITEMS(seq); local
2256 PyObject* tensor = seq_array[i];
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc993 auto seq_array = sequence_lengths->template flat<int>().data(); in ShouldUsePaddedIO() local
996 if (seq_array[i] != model_shapes.max_seq_length) { in ShouldUsePaddedIO()