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

  /external/tensorflow/tensorflow/compiler/xla/service/
service.cc 166 << "Requested more replicas than there are devices.";
415 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*backend, device_handles[i]));
416 CHECK_EQ(replicas.size(), arguments[i].size());
417 for (int64 replica = 0; replica < replicas.size(); ++replica) {
418 device_assignment(replica, i) = replicas[replica]->device_ordinal();
423 // Stream executors for the replicas of the current computation.
424 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*backend, device_handles[i]));
425 CHECK_EQ(replicas.size(), arguments[i].size())
977 std::vector<se::StreamExecutor*> replicas; local
1163 std::vector<se::StreamExecutor*> replicas; local
    [all...]
service.h 56 // Set the default number of replicas to use when compiling replicated
119 // When N device handles are requested and the number of replicas is R, at
122 // the first set of replicas, and the next R devices to the second set of
123 // replicas, etc. Each returned device handle represents the device with the
272 // Returns the stream executors assigned to the replicas represented by the
274 // represents a set of physical devices for the replicas.
275 StatusOr<std::vector<se::StreamExecutor*>> Replicas(
  /external/tensorflow/tensorflow/python/keras/engine/
distributed_training_utils.py 494 'replicas (%s) in the distribution strategy.' % (
604 """Build an updated model on replicas.
620 model: Model to be replicated across Replicas
    [all...]

Completed in 105 milliseconds