Searched refs:input_specs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils.py | 82 input_specs = nest.pack_sequence_as(structure=inputs,
86 input_signature = [input_specs] if len(input_specs) > 1 else input_specs
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 597 def build_infeed_from_input_specs(self, input_specs, execution_mode): argument
627 input_specs = []
631 input_specs.append(
635 return input_specs
679 def build_infeed_from_input_specs(self, input_specs, execution_mode): argument
688 for spec in input_specs:
699 infeed_tensors, [spec.shape for spec in input_specs],
714 def __init__(self, input_specs): argument
715 self._input_specs = input_specs
764 input_specs = []
[all …]
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_v1_in_v2.py | 101 input_names, input_specs = zip(*signature_def.inputs.items())
105 for inp in input_specs],
|