Home
last modified time | relevance | path

Searched refs:num_cores_per_replica (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.cc45 Status CoreOutOfRangeError(int core, int num_cores_per_replica) { in CoreOutOfRangeError() argument
48 "; num_cores_per_replica=", num_cores_per_replica); in CoreOutOfRangeError()
53 const string& device_name, int num_cores_per_replica, in ParseShardingFromDevice() argument
72 if (core < 0 || core >= num_cores_per_replica) { in ParseShardingFromDevice()
73 return CoreOutOfRangeError(core, num_cores_per_replica); in ParseShardingFromDevice()
81 const NodeDef& node_def, int num_cores_per_replica) { in ParseShardingFromDevice() argument
85 return ParseShardingFromDevice(device_name, num_cores_per_replica, sharding); in ParseShardingFromDevice()
89 const Node& node, int num_cores_per_replica) { in ParseShardingFromDevice() argument
96 return ParseShardingFromDevice(device_name, num_cores_per_replica, sharding); in ParseShardingFromDevice()
Dsharding_util.h37 const string& device_name, int num_cores_per_replica,
41 const Node& node, int num_cores_per_replica);
44 const NodeDef& node_def, int num_cores_per_replica);
Dtf2xla_util_test.cc235 const int num_cores_per_replica = 4; in TEST() local
243 auto parse_status = ParseShardingFromDevice(*c_node, num_cores_per_replica); in TEST()
251 parse_status = ParseShardingFromDevice(*c_node, num_cores_per_replica); in TEST()
258 parse_status = ParseShardingFromDevice(*a_node, num_cores_per_replica); in TEST()
/external/tensorflow/tensorflow/python/tpu/
Dtpu_context.py215 use_tpu and config.tpu_config.num_cores_per_replica)
217 num_cores_per_replica = config.tpu_config.num_cores_per_replica
220 num_cores_per_replica]
294 str(self._config.tpu_config.num_cores_per_replica))
365 num_cores_per_replica = self._config.tpu_config.num_cores_per_replica
366 if num_cores_per_replica > num_cores_in_system:
371 'in the system: {}'.format(num_cores_per_replica,
374 if num_cores_in_system % num_cores_per_replica != 0:
379 num_cores_in_system, num_cores_per_replica))
381 return num_cores_in_system // num_cores_per_replica
[all …]
Dtpu_config.py119 num_cores_per_replica=None, argument
144 if num_cores_per_replica is None:
149 if num_cores_per_replica is not None:
150 if num_cores_per_replica not in [1, 2, 4, 8, 16]:
153 str(num_cores_per_replica)))
180 num_cores_per_replica=num_cores_per_replica,
Dtpu_feed.py846 for logical_core in xrange(self._device_assignment.num_cores_per_replica):
893 if dims.prod() != self._device_assignment.num_cores_per_replica:
897 "= {})".format(dims, self._device_assignment.num_cores_per_replica))
Dtpu_config_test.py91 tpu_config_lib.TPUConfig(num_cores_per_replica=7)
Ddevice_assignment.py109 def num_cores_per_replica(self): member in DeviceAssignment
Dtpu.py735 device_assignment.num_cores_per_replica)
738 device_assignment.num_cores_per_replica
Dtpu_estimator.py2322 config.tpu_config.num_cores_per_replica):
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_compile_ops.cc90 int num_cores_per_replica = in Compile() local
91 config.num_cores_per_replica() ? config.num_cores_per_replica() : 1; in Compile()
92 TF_RET_CHECK(num_cores_per_replica == 1); in Compile()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TPUReplicateMetadata.pbtxt11 name: "num_cores_per_replica"
37 DEPRECATED. Use num_cores_per_replica instead.
Dapi_def_TPUReplicate.pbtxt45 name: "num_cores_per_replica"
69 [replica, num_cores_per_replica, mesh_dimension] that maps the coordinates
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py221 if self._device_assignment.num_cores_per_replica != 1:
545 self._device_assignment.num_cores_per_replica)
548 return models_per_host * self._device_assignment.num_cores_per_replica
555 self._device_assignment.num_cores_per_replica)
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt.proto35 int32 num_cores_per_replica = 2; field
46 // devices. There are num_cores_per_replica computations, and each one will be
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt10 name: "num_cores_per_replica"
Dtensorflow.raw_ops.pbtxt3849 …nstants\', \'computation\', \'num_replicas\', \'output_types\', \'num_cores_per_replica\', \'topol…
3853 …argspec: "args=[\'num_replicas\', \'num_cores_per_replica\', \'topology\', \'use_tpu\', \'device_a…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt10 name: "num_cores_per_replica"
Dtensorflow.raw_ops.pbtxt3849 …nstants\', \'computation\', \'num_replicas\', \'output_types\', \'num_cores_per_replica\', \'topol…
3853 …argspec: "args=[\'num_replicas\', \'num_cores_per_replica\', \'topology\', \'use_tpu\', \'device_a…
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt82135 name: "num_cores_per_replica"
82240 name: "num_cores_per_replica"
82326 name: "num_cores_per_replica"
82387 name: "num_cores_per_replica"
Dops_history.v1.pbtxt82762 name: "num_cores_per_replica"
82867 name: "num_cores_per_replica"
82953 name: "num_cores_per_replica"
83014 name: "num_cores_per_replica"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt39917 name: "num_cores_per_replica"
40003 name: "num_cores_per_replica"