Home
last modified time | relevance | path

Searched refs:flat_sequence (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/python/data/util/
Dnest.py159 def pack_sequence_as(structure, flat_sequence): argument
177 if not (is_sequence(flat_sequence) or isinstance(flat_sequence, list)):
181 if len(flat_sequence) != 1:
183 % len(flat_sequence))
184 return flat_sequence[0]
187 if len(flat_structure) != len(flat_sequence):
191 % (len(flat_structure), len(flat_sequence), structure, flat_sequence))
193 _, packed = _packed_nest_with_indices(structure, flat_sequence, 0)
474 return pack_sequence_as(structure=shallow_tree, flat_sequence=results)
/external/tensorflow/tensorflow/python/util/
Dnest.py387 def pack_sequence_as(structure, flat_sequence, expand_composites=False): argument
420 if not is_seq(flat_sequence):
424 if len(flat_sequence) != 1:
426 % len(flat_sequence))
427 return flat_sequence[0]
430 final_index, packed = _packed_nest_with_indices(structure, flat_sequence,
432 if final_index < len(flat_sequence):
436 if len(flat_structure) != len(flat_sequence):
440 (len(flat_structure), len(flat_sequence), structure, flat_sequence))
1034 return pack_sequence_as(structure=shallow_tree, flat_sequence=results)
/external/tensorflow/tensorflow/python/ops/
Drnn.py302 structure=zero_output, flat_sequence=final_output)
304 structure=state, flat_sequence=final_state)
346 results = [nest.pack_sequence_as(structure=input_, flat_sequence=flat_result)
662 inputs = nest.pack_sequence_as(structure=inputs, flat_sequence=flat_input)
766 flat_sequence=flat_zero_output)
828 input_t = nest.pack_sequence_as(structure=inputs, flat_sequence=input_t)
893 structure=cell.output_size, flat_sequence=final_outputs)
1114 flat_sequence=flat_state)
1138 flat_sequence=flat_emit_ta)
1143 flat_sequence=flat_zero_emit)
[all …]
Dcustom_gradient.py253 structure=result, flat_sequence=all_tensors[:len(flat_result)])
Dctc_ops.py1070 pack_elems = lambda x: nest.pack_sequence_as(structure=elems, flat_sequence=x)
1072 pack = lambda x: nest.pack_sequence_as(structure=initial, flat_sequence=x)
Dcontrol_flow_ops.py2024 merges = nest.pack_sequence_as(structure=orig_res_t, flat_sequence=merges,
2905 flat_sequence=merge_vars_with_tensor_arrays,
2920 flat_sequence=vars_for_body_with_tensor_arrays,
3009 flat_sequence=exit_vars_with_tensor_arrays,
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py799 flat_sequence = nest.flatten(sequence, expand_composites=True)
802 for item in flat_sequence]
805 def pack_sequence_as(structure, flat_sequence): argument
819 flat_sequence = list(flat_sequence)
821 if len(flattened_structure) != len(flat_sequence):
823 for i in range(len(flat_sequence)):
825 flat_sequence[i] = tensor_array_ops.build_ta_with_new_flow(
826 old_ta=flattened_structure[i], flow=flat_sequence[i])
827 return nest.pack_sequence_as(structure, flat_sequence, expand_composites=True)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nest.pbtxt21 …argspec: "args=[\'structure\', \'flat_sequence\', \'expand_composites\'], varargs=None, keywords=N…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nest.pbtxt21 …argspec: "args=[\'structure\', \'flat_sequence\', \'expand_composites\'], varargs=None, keywords=N…
/external/tensorflow/tensorflow/python/keras/saving/
Dsaving_utils.py83 flat_sequence=flat_input_specs)
/external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/ops/
Dseq2seq.py406 structure=encoder_state, flat_sequence=state_list)
535 structure=zero_state, flat_sequence=state_list)
922 structure=encoder_state, flat_sequence=state_list)
1050 structure=encoder_state, flat_sequence=state_list)
/external/tensorflow/tensorflow/contrib/compiler/
Dxla.py331 structure=inputs, flat_sequence=flat_inputs)
380 structure=outputs, flat_sequence=output_tensors)
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn.py257 inputs = nest.pack_sequence_as(structure=inputs, flat_sequence=flat_input)
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dhead.py206 flat_sequence=[tensor for _, _, tensor in numbered_state])
/external/tensorflow/tensorflow/python/eager/
Dfunction.py1128 structure=inputs, flat_sequence=flat_inputs)
1171 flat_sequence=flatten_inputs)
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py864 flat_sequence=flat_replicated_inputs[:flat_input_arity])