Home
last modified time | relevance | path

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

12

/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.cc237 const int num_cores_per_replica = 4; in TEST() local
245 auto parse_status = ParseShardingFromDevice(*c_node, num_cores_per_replica); in TEST()
253 parse_status = ParseShardingFromDevice(*c_node, num_cores_per_replica); in TEST()
260 parse_status = ParseShardingFromDevice(*a_node, num_cores_per_replica); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_rewrite.mlir8 …() {_tpu_replicate = "cluster0", device = "", func = @empty_func, num_cores_per_replica = 1, step_…
23 …() {_tpu_replicate = "cluster0", device = "", func = @empty_func, num_cores_per_replica = 1, step_…
38 // expected-error@+1 {{requires attribute 'num_cores_per_replica'}}
53 // expected-error@+1 {{requires attribute 'num_cores_per_replica'}}
54 …() {_tpu_replicate = "cluster0", device = "", func = @empty_func, num_cores_per_replica = "", step…
69 …() {_tpu_replicate = "cluster0", device = "", func = @empty_func, num_cores_per_replica = 1, paddi…
84 …() {_tpu_replicate = "cluster0", device = "", func = @empty_func, num_cores_per_replica = 1, step_…
99 …() {_tpu_replicate = "cluster0", device = "", func = @empty_func, num_cores_per_replica = 1, step_…
114 …() {_tpu_replicate = "cluster0", device = "", func = @empty_func, num_cores_per_replica = 1, step_…
129 …() {_tpu_replicate = "cluster0", device = "", func = @empty_func, num_cores_per_replica = 1, step_…
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc157 tf_device::LaunchFuncOp op, int num_replicas, int num_cores_per_replica, in SetMetadataProtoFromLaunchFuncOp() argument
160 metadata->set_num_cores_per_replica(num_cores_per_replica); in SetMetadataProtoFromLaunchFuncOp()
255 int num_cores_per_replica, in BuildCompileOp() argument
264 launch_func, num_replicas, num_cores_per_replica, &metadata))) in BuildCompileOp()
433 int num_cores_per_replica = num_cores_per_replica_attr.getInt(); in Rewrite() local
439 devices, num_replicas, num_cores_per_replica, &compilation_device, in Rewrite()
449 BuildCompileOp(launch_func, num_replicas, num_cores_per_replica, in Rewrite()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DTPUReplicateMetadata.pbtxt9 name: "num_cores_per_replica"
70 name: "num_cores_per_replica"
138 name: "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/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc153 Devices devices, int num_replicas, int num_cores_per_replica, in GetTPUCompilationAndExecutionDevices() argument
156 if (num_cores_per_replica != 1) in GetTPUCompilationAndExecutionDevices()
159 num_cores_per_replica); in GetTPUCompilationAndExecutionDevices()
Dtpu_rewrite_device_util.h56 int num_cores_per_replica, std::string* compilation_device,
/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.
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.estimator.tpu.-t-p-u-config.pbtxt27 name: "num_cores_per_replica"
Dtensorflow.tpu.experimental.-device-assignment.pbtxt10 name: "num_cores_per_replica"
Dtensorflow.raw_ops.pbtxt4453 …argspec: "args=[\'num_replicas\', \'num_cores_per_replica\', \'topology\', \'use_tpu\', \'device_a…
/external/tensorflow/tensorflow/core/profiler/protobuf/
Dop_stats.proto82 // If there is no model parallelism, then num_cores_per_replica = 1
83 int32 num_cores_per_replica = 10; field
Doverview_page.proto137 // If there is no model parallelism, then num_cores_per_replica = 1
138 int32 num_cores_per_replica = 10; field
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt10 name: "num_cores_per_replica"
Dtensorflow.raw_ops.pbtxt4453 …argspec: "args=[\'num_replicas\', \'num_cores_per_replica\', \'topology\', \'use_tpu\', \'device_a…
/external/tensorflow/tensorflow/python/tpu/
Dtpu_feed.py858 for logical_core in xrange(self._device_assignment.num_cores_per_replica):
909 if dims.prod() != self._device_assignment.num_cores_per_replica:
913 "= {})".format(dims, self._device_assignment.num_cores_per_replica))
Ddevice_assignment.py111 def num_cores_per_replica(self): member in DeviceAssignment
Dtpu.py322 for logical_core in range(device_assignment.num_cores_per_replica):
1035 device_assignment.num_cores_per_replica)
/external/tensorflow/tensorflow/core/profiler/convert/
Dop_stats_to_overview_page.cc174 re.set_num_cores_per_replica(run_environment.num_cores_per_replica()); in ComputeRunEnvironment()
/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/core/protobuf/tpu/
Dcompile_metadata.proto78 int32 num_cores_per_replica = 4; field
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py255 for logical_core in range(device_assignment.num_cores_per_replica):
679 self._device_assignment.num_cores_per_replica)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td468 DefaultValuedAttr<I64Attr, "1">:$num_cores_per_replica,

12