OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:replicas
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
service.cc
157
<< "Requested more
replicas
than there are devices.";
514
// Stream executors for the
replicas
of the current computation.
515
TF_ASSIGN_OR_RETURN(auto
replicas
,
Replicas
(*backend, device_handles[i]));
516
for (int64 replica = 0; replica <
replicas
.size(); ++replica) {
518
backend->BorrowStream(
replicas
[replica]));
556
// All
replicas
share the same device address for the result allocation,
557
// so only one of the
replicas
need to register the result handle.
636
TF_ASSIGN_OR_RETURN(auto
replicas
,
637
Replicas
(*backend, SingleComputationDeviceHandle()))
1068
std::vector<se::StreamExecutor*>
replicas
;
local
1614
std::vector<perftools::gputools::StreamExecutor*>
replicas
;
local
[
all
...]
/external/autotest/site_utils/
check_slave_db_delay.py
11
Readonly
replicas
of TKO database, passed in by option --
replicas
.
72
parser.add_argument('-r', '--
replicas
', nargs='+',
73
help='IP addresses of readonly
replicas
of TKO.')
98
for replica in options.
replicas
:
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
device_assignment.py
107
"""Computes a nested dict which maps task and logical core to
replicas
."""
127
for core,
replicas
in core_to_replicas.items():
128
core_to_sorted_replicas[core] = sorted(
replicas
)
159
"""The number of
replicas
of the computation."""
193
A sorted list of the
replicas
that are attached to that task and
247
num_replicas: The number of computation
replicas
to run. The
replicas
will
259
ValueError: If computation's
replicas
cannot fit into the TPU topology.
311
"requested {}
replicas
but only {}
replicas
with shape {} and
[
all
...]
tpu.py
194
`[[]]`), indexed by `[replica_num][input_num]`. All
replicas
must
203
replicas
is equal to the number of cores in the TPU system.
208
ValueError: If all
replicas
do not have equal numbers of input tensors.
235
# No
replicas
? Nothing to do.
242
# Verifies that all
replicas
have matching numbers and types of inputs
247
raise ValueError("
Replicas
must have the same number of inputs. "
254
"
Replicas
must have matching input types. Replica 0 had "
281
replicas
= [inputs[replica][i] for replica in xrange(num_replicas)]
283
tpu_ops.tpu_replicated_input(
replicas
, name="input{}".format(i)))
/external/tensorflow/tensorflow/contrib/estimator/python/estimator/
replicate_model_fn.py
74
the CPU.
Replicas
of operations are placed on each individual GPU.
176
the CPU.
Replicas
of operations are placed on each individual GPU.
708
# `i` is `[1.. number_of_towers]`. We are going to add values from
replicas
716
replicas
= [
720
] # `
replicas
` doesn't contain the first-tower variable.
723
math_ops.add_n(
replicas
))
726
for replica in
replicas
:
Completed in 1282 milliseconds