HomeSort by relevance Sort by last modified time
    Searched refs:input_specs (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/python/keras/saving/
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/
keras_support.py 574 the model based on the returned `input_specs` from `TPUInfeedInstance`.
597 def build_infeed_from_input_specs(self, input_specs, execution_mode):
604 input_specs: TODO(saeta): Document me!
627 input_specs = []
629 # metadata (`input_specs`); all replicas have the same type and shape.
631 input_specs.append(
635 return input_specs
679 def build_infeed_from_input_specs(self, input_specs, execution_mode):
688 for spec in input_specs:
699 infeed_tensors, [spec.shape for spec in input_specs],
    [all...]
  /external/tensorflow/tensorflow/python/saved_model/
load_v1_in_v2.py 101 input_names, input_specs = zip(*signature_def.inputs.items())
105 for inp in input_specs],

Completed in 835 milliseconds