Searched refs:replicated (Results 1 – 25 of 79) sorted by relevance
1234
4 summary: "Connects N inputs to an N-way replicated TPU computation."
4 summary: "Connects outputs of an N-way replicated computation to N outputs."
29 summary: "An Op to permute tensors across replicated TPU instances."
46 summary: "Metadata indicaitng how the TPU computation should be replicated."
19 and the values of `input` are replicated `multiples[i]` times along the 'i'th
30 summary: "An Op to sum inputs across replicated TPU instances."
99 summary: "Runs replicated computations on a distributed TPU system."
35 # Changes to the Android srcs here should be replicated in
51 @* The top row of a 2d array is replicated for pad_size times at the top120 @* The left column of a 2d array is replicated for pad_size times at the left256 @* The left column of a 2d array is replicated for pad_size times at the left384 @* The right column of a 2d array is replicated for pad_size times at the right530 @* The right column of a 2d array is replicated for pad_size times at the right
41 @* All the functions here are replicated from ih264_inter_pred_filters.c
20 // As many replicas as there are in the replicated computation.45 // Describes how replicated computation instances should be assigned to
139 ; ...and again with the lower bits of the replicated constant.161 ; ...and again with the lower bits of the replicated constant.
1 Scenario: A class that is replicated using a CloneVisitor should be equal to the source
305 // Handle given to a user that represents a replicated virtual device. Each306 // replicated device represents N physical devices for execution where N is the340 // represents the device ids assigned to a set of replicated computations.577 // This sharding is replicated across all devices (implies maximal,639 // indicates whether the corresponding leaf buffer is replicated.641 // If this field is empty, it means no buffer is replicated. Otherwise, the
36 # Change to compile flags should be replicated into bazel build file
31 // range will be summed (with the gradients replicated in the backward
82 replicated VF*UF times to handle scalarized and predicated instructions.134 CFG. A VPRegionBlock may indicate that its contents are to be replicated a145 instructions; e.g., cloned once, replicated multiple times or widened
135 creates one single graph that is replicated across all the cores in the Cloud163 - The `model_fn` models the computation which will be replicated and distributed
119 ; ~(C >>s Y) --> ~C >>u Y (when inverting the replicated sign bits)151 ; ~(C >>u Y) --> ~C >>s Y (when inverting the replicated sign bits)
290 KEYWORD(replicated); in LexIdentifier()
1727 const auto& replicated = in ParseInstructionRhs() local1729 if (leaf_count != replicated.size()) { in ParseInstructionRhs()1733 replicated.size(), " elements.")); in ParseInstructionRhs()1735 instruction->set_parameter_replicated_at_leaf_buffers(replicated); in ParseInstructionRhs()1800 bool replicated = false; in ParseSingleSharding() local1810 replicated = true; in ParseSingleSharding()1859 if (replicated) { in ParseSingleSharding()
35 are replicated across all local GPUs. In a multi-machine setting, some are37 one parameter server. Computation operations are replicated across all GPUs of