Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/util/
Dnest.py157 def pack_sequence_as(structure, flat_sequence): argument
175 if not (is_sequence(flat_sequence) or isinstance(flat_sequence, list)):
179 if len(flat_sequence) != 1:
181 % len(flat_sequence))
182 return flat_sequence[0]
185 if len(flat_structure) != len(flat_sequence):
189 % (len(flat_structure), len(flat_sequence), structure, flat_sequence))
191 _, packed = _packed_nest_with_indices(structure, flat_sequence, 0)
472 return pack_sequence_as(structure=shallow_tree, flat_sequence=results)
/external/tensorflow/tensorflow/python/util/
Dnest.py431 def _pack_sequence_as(structure, flat_sequence, expand_composites, argument
440 if not is_seq(flat_sequence):
444 .format(truncate(flat_sequence, 100), type(flat_sequence)))
447 if len(flat_sequence) != 1:
452 type(structure), truncate(structure, 100), type(flat_sequence),
453 len(flat_sequence), truncate(flat_sequence, 100)))
454 return flat_sequence[0]
457 final_index, packed = _packed_nest_with_indices(structure, flat_sequence,
459 if final_index < len(flat_sequence):
463 if len(flat_structure) != len(flat_sequence):
[all …]
/external/tensorflow/tensorflow/python/ops/
Drnn.py311 structure=zero_output, flat_sequence=final_output)
313 structure=state, flat_sequence=final_state)
354 nest.pack_sequence_as(structure=input_, flat_sequence=flat_result)
698 inputs = nest.pack_sequence_as(structure=inputs, flat_sequence=flat_input)
802 structure=cell.output_size, flat_sequence=flat_zero_output)
865 input_t = nest.pack_sequence_as(structure=inputs, flat_sequence=input_t)
931 structure=cell.output_size, flat_sequence=final_outputs)
1148 state = nest.pack_sequence_as(structure=state, flat_sequence=flat_state)
1175 structure=emit_structure, flat_sequence=flat_emit_ta)
1181 structure=emit_structure, flat_sequence=flat_zero_emit)
[all …]
Dcustom_gradient.py402 structure=result, flat_sequence=all_tensors[:flat_result_len])
Dctc_ops.py1319 pack_elems = lambda x: nest.pack_sequence_as(structure=elems, flat_sequence=x)
1321 pack = lambda x: nest.pack_sequence_as(structure=initial, flat_sequence=x)
Dcontrol_flow_ops.py1291 structure=orig_res_t, flat_sequence=merges, expand_composites=True)
2155 flat_sequence=merge_vars_with_tensor_arrays,
2170 flat_sequence=vars_for_body_with_tensor_arrays,
2259 flat_sequence=exit_vars_with_tensor_arrays,
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py1084 flat_sequence = nest.flatten(sequence, expand_composites=True)
1087 for item in flat_sequence]
1091 def pack_sequence_as(structure, flat_sequence): argument
1105 flat_sequence = list(flat_sequence)
1107 if len(flattened_structure) != len(flat_sequence):
1109 for i in range(len(flat_sequence)):
1111 flat_sequence[i] = tensor_array_ops.build_ta_with_new_flow(
1112 old_ta=flattened_structure[i], flow=flat_sequence[i])
1113 return nest.pack_sequence_as(structure, flat_sequence, expand_composites=True)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nest.pbtxt21 …argspec: "args=[\'structure\', \'flat_sequence\', \'expand_composites\'], varargs=None, keywords=N…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nest.pbtxt21 …argspec: "args=[\'structure\', \'flat_sequence\', \'expand_composites\'], varargs=None, keywords=N…
/external/tensorflow/tensorflow/python/compiler/xla/
Dxla.py355 structure=inputs, flat_sequence=flat_inputs)
404 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/python/keras/saving/
Dsaving_utils.py97 flat_sequence=flat_input_specs)
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py1170 flat_sequence=flat_replicated_inputs[:flat_input_arity])
/external/tensorflow/tensorflow/python/eager/
Dfunction.py2254 structure=inputs, flat_sequence=flat_inputs, expand_composites=False)
2304 flat_sequence=flatten_inputs,