| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | sharding_util.cc | 45 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()
|
| D | sharding_util.h | 37 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);
|
| D | tf2xla_util_test.cc | 235 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/ |
| D | tpu_context.py | 215 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 …]
|
| D | tpu_config.py | 119 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,
|
| D | tpu_feed.py | 846 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))
|
| D | tpu_config_test.py | 91 tpu_config_lib.TPUConfig(num_cores_per_replica=7)
|
| D | device_assignment.py | 109 def num_cores_per_replica(self): member in DeviceAssignment
|
| D | tpu.py | 735 device_assignment.num_cores_per_replica)
738 device_assignment.num_cores_per_replica
|
| D | tpu_estimator.py | 2322 config.tpu_config.num_cores_per_replica):
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 90 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/ |
| D | api_def_TPUReplicateMetadata.pbtxt | 11 name: "num_cores_per_replica"
37 DEPRECATED. Use num_cores_per_replica instead.
|
| D | api_def_TPUReplicate.pbtxt | 45 name: "num_cores_per_replica"
69 [replica, num_cores_per_replica, mesh_dimension] that maps the coordinates
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 221 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/ |
| D | xrt.proto | 35 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/ |
| D | tensorflow.tpu.experimental.-device-assignment.pbtxt | 10 name: "num_cores_per_replica"
|
| D | tensorflow.raw_ops.pbtxt | 3849 …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/ |
| D | tensorflow.tpu.experimental.-device-assignment.pbtxt | 10 name: "num_cores_per_replica"
|
| D | tensorflow.raw_ops.pbtxt | 3849 …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/ |
| D | ops_history.v2.pbtxt | 82135 name: "num_cores_per_replica"
82240 name: "num_cores_per_replica"
82326 name: "num_cores_per_replica"
82387 name: "num_cores_per_replica"
|
| D | ops_history.v1.pbtxt | 82762 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/ |
| D | ops.pbtxt | 39917 name: "num_cores_per_replica"
40003 name: "num_cores_per_replica"
|