| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | sharding_util.cc | 49 Status CoreOutOfRangeError(int core, int num_cores_per_replica) { in CoreOutOfRangeError()
57 const string& device_name, int num_cores_per_replica, in ParseShardingFromDevice()
89 const NodeDef& node_def, int num_cores_per_replica, bool add_metadata) { in ParseShardingFromDevice()
101 const Node& node, int num_cores_per_replica, bool add_metadata) { in ParseShardingFromDevice()
116 const Edge& edge, int num_cores_per_replica, bool add_metadata) { in ParseShardingFromEdgeSource()
|
| D | tf2xla_util_test.cc | 237 const int num_cores_per_replica = 4; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 303 tf_device::ClusterFuncOp op, int num_replicas, int num_cores_per_replica, in SetMetadataProtoFromClusterFuncOp()
351 int num_cores_per_replica, llvm::StringRef compilation_device, in BuildCompileOp()
419 const int num_cores_per_replica = tpu_devices.front().size(); in AssignDevicesToReplicate() local
474 const int num_cores_per_replica = tpu_devices.front().size(); in BuildParallelExecuteOp() local
582 int num_cores_per_replica = num_cores_per_replica_attr.getInt(); in Rewrite() local
|
| D | tpu_reorder_replicate_and_partitioned_inputs.cc | 51 size_t num_cores_per_replica = first_partitioned_input.getNumOperands(); in ReorderReplicateAndPartitionedInputs() local
|
| D | tpu_cluster_formation.cc | 343 int num_cores_per_replica) { in ReplicateCluster()
568 int num_cores_per_replica = 1; in FormClustersInBlock() local
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_compile_ops.cc | 93 int num_cores_per_replica) { in CompilationCacheKey()
179 int num_cores_per_replica = in Compute() local
|
| D | xrt_compile_ops.cc | 56 int num_cores_per_replica, xla::DeviceAssignment* device_assignment) { in GenerateXlaDeviceAssignment()
127 int num_cores_per_replica = std::max<int>(config.num_cores_per_replica(), 1); in Compile() local
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | compile_metadata.proto | 81 int32 num_cores_per_replica = 4; field
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 188 int num_replicas, int num_cores_per_replica, in GetFullMeshTPUExecutionDeviceAssignment()
347 int num_replicas, int num_cores_per_replica, in GetGeneralTPUExecutionDeviceAssignment()
450 Devices devices, int num_replicas, int num_cores_per_replica, in GetTPUCompilationAndExecutionDevices()
|
| D | xla_sharding_util.cc | 211 const int num_cores_per_replica, in ExtractInputsForLogicalDevices()
312 const int num_cores_per_replica, in ParseAndValidateOutputSharding()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 490 Status ValidateCoreNumber(int64 core, int64 num_cores_per_replica) { in ValidateCoreNumber()
1196 const int num_cores_per_replica, in ParseAndValidateSharding()
1261 ParseInputShardingFromAdjacentNode(const int num_cores_per_replica, in ParseInputShardingFromAdjacentNode()
1299 const int num_cores_per_replica, const std::string& arg_node_name, in ParseAndValidateShardingFromNeighbors()
1444 int num_cores_per_replica, in ParseDeviceAssignmentAttr()
1524 int num_replicas, int num_cores_per_replica, in BuildGeneralDeviceAssignment()
1564 int num_cores_per_replica, const string& topology_attr, in BuildDeviceAssignment()
1814 int num_cores_per_replica) { in ValidateCoreNumbers()
1824 const Node& n, int num_cores_per_replica, FunctionLibraryRuntime* flr, in InferXlaShardingFromNeighbors()
1912 int num_cores_per_replica, const ParameterInfo& params_info, in AssignArgsAndRetvalsToCores()
[all …]
|
| D | encapsulate_tpu_computations_pass.cc | 166 int64 num_cores_per_replica = 1LL; in RewriteSubgraph() local
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | op_stats.proto | 86 int32 num_cores_per_replica = 10; field
|
| D | overview_page.proto | 187 int32 num_cores_per_replica = 10; field
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | device_assignment.py | 113 def num_cores_per_replica(self) -> int: member in DeviceAssignment
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt.proto | 36 int32 num_cores_per_replica = 2; field
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_program_group.cc | 301 int num_cores_per_replica = in CompileAndBuild() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_ops_c_api.h | 73 int32_t num_cores_per_replica; member
|