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

  /external/tensorflow/tensorflow/python/tpu/
functional.py 21 from tensorflow.python.tpu.ops import tpu_ops
23 TPUPartitionedCall = tpu_ops.tpu_partitioned_call # pylint: disable=invalid-name
tpu_embedding_gradient.py 27 from tensorflow.python.tpu.ops import tpu_ops
119 new_activations[feature] = tpu_ops.tpu_embedding_activations(
tpu_feed.py 35 from tensorflow.python.tpu.ops import tpu_ops
488 return tpu_ops.infeed_dequeue_tuple(
491 return tpu_ops.infeed_dequeue_tuple(
534 return tpu_ops.infeed_enqueue_tuple(
541 return tpu_ops.infeed_enqueue_tuple(
787 values = tpu_ops.infeed_dequeue_tuple(
    [all...]
tpu_embedding.py 39 from tensorflow.python.tpu.ops import tpu_ops
653 return tpu_ops.enqueue_tpu_embedding_sparse_tensor_batch(
699 recv_activations = tpu_ops.recv_tpu_embedding_activations(
738 return tpu_ops.send_tpu_embedding_gradients(
    [all...]
tpu_optimizer.py 27 from tensorflow.python.tpu.ops import tpu_ops
145 Calls tpu_ops.cross_replica_sum() to sum gradient contributions across
169 summed_grads_and_vars.append((tpu_ops.cross_replica_sum(
tpu.py 40 from tensorflow.python.tpu.ops import tpu_ops
102 return tpu_ops.configure_distributed_tpu(embedding_config=config_string)
108 shutdown_distributed_tpu = tpu_ops.shutdown_distributed_tpu()
195 handle = tpu_ops.tpu_replicated_input(
    [all...]
session_support.py 34 from tensorflow.python.tpu.ops import tpu_ops
86 heartbeat_ops.append(tpu_ops.worker_heartbeat(request_placeholder))
tensor_tracer.py 44 from tensorflow.python.tpu.ops import tpu_ops
660 self._replica_id = tpu_ops.tpu_replicated_input(
    [all...]
tpu_estimator.py 79 from tensorflow.python.tpu.ops import tpu_ops
    [all...]
  /external/tensorflow/tensorflow/contrib/tpu/python/ops/
tpu_ops.py 22 from tensorflow.python.tpu.ops.tpu_ops import *
  /external/tensorflow/tensorflow/contrib/tpu/
__init__.py 69 from tensorflow.contrib.tpu.python.ops.tpu_ops import *
  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
keras_support.py 58 from tensorflow.contrib.tpu.python.ops import tpu_ops
318 concat_tensor = tpu_ops.cross_replica_sum(local_tensor_with_holes)
352 return [tpu_ops.cross_replica_sum(grad) / num_shards for grad in grads]
698 tpu_ops.infeed_enqueue_tuple(
842 tpu_ops.infeed_enqueue_tuple(
    [all...]
  /external/tensorflow/tensorflow/python/distribute/
tpu_strategy.py 48 from tensorflow.python.tpu.ops import tpu_ops
449 return tpu_ops.cross_replica_sum(value)

Completed in 675 milliseconds