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

  /external/tensorflow/tensorflow/python/tpu/
tpu_feed.py 548 sharded_inputs,
553 sharded_inputs is a list, one for each shard, of lists of
554 Tensors. sharded_inputs[0] is the tuple of Tensors to use to feed
561 compatible with the types and shapes of sharded_inputs, an error
565 sharded_inputs: a list of lists of Tensors. The length of the outer list
582 shapes of the elements of sharded_inputs are not compatible with the
583 frozen configuration; or if the shapes of the elements of sharded_inputs
587 types of the elements of sharded_inputs are not compatible with the
588 frozen configuration; or if the types of the elements of sharded_inputs
591 self.set_configuration_from_sharded_input_tensors(sharded_inputs)
    [all...]
  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
keras_support.py 619 def __init__(self, sharded_inputs):
620 self._sharded_inputs = sharded_inputs
676 sharded_inputs = self._split_tensors(inputs)
677 return self.NumpyInfeedInstance(sharded_inputs)
    [all...]

Completed in 670 milliseconds