Searched refs:is_ragged_input (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | wrappers.py | 220 is_ragged_input = row_lengths is not None
233 input_length=row_lengths[0] if is_ragged_input else input_length,
238 lambda output: K.maybe_convert_to_ragged(is_ragged_input, output,
244 is_ragged_input = nest.map_structure(
246 is_ragged_input = nest.flatten(is_ragged_input)
247 if all(is_ragged_input):
254 elif any(is_ragged_input):
337 is_ragged_input = nest.map_structure(
339 is_ragged_input = generic_utils.to_list(nest.flatten(is_ragged_input))
340 if batch_size and not self._always_use_reshape or any(is_ragged_input):
|
| D | recurrent_v2.py | 425 is_ragged_input = (row_lengths is not None)
426 self._validate_args_if_ragged(is_ragged_input, mask)
439 if is_ragged_input or not self._could_use_gpu_kernel:
468 output = K.maybe_convert_to_ragged(is_ragged_input, outputs, row_lengths)
1151 is_ragged_input = (row_lengths is not None)
1152 self._validate_args_if_ragged(is_ragged_input, mask)
1165 if is_ragged_input or not self._could_use_gpu_kernel:
1280 output = K.maybe_convert_to_ragged(is_ragged_input, outputs, row_lengths)
|
| D | recurrent.py | 740 is_ragged_input = (row_lengths is not None)
741 self._validate_args_if_ragged(is_ragged_input, mask)
826 output = K.maybe_convert_to_ragged(is_ragged_input, outputs, row_lengths)
879 def _validate_args_if_ragged(self, is_ragged_input, mask): argument
880 if not is_ragged_input:
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 6592 def maybe_convert_to_ragged(is_ragged_input, output, nested_row_lengths): argument
6594 if not is_ragged_input:
|