- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 137 for core_id (0.35 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_reorder_replicate_and_partitioned_inputs.cc
for (auto operand : replicated_input.getInputs()) { Operation* pi = operand.getDefiningOp(); for (unsigned core_id = 0; core_id < num_cores_per_replica; ++core_id) { const auto pi_operand = packed_input ? pi->getOperand(0) : pi->getOperand(core_id); operands_per_replica_per_core[core_id].push_back(pi_operand); } } // Create new `tf.TPUReplicatedInput` ops feeding into one
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 24 23:08:55 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
} return mlir::success(); } namespace { bool IsAssignedToLogicalDevice(const int core_id, const xla::OpSharding& sharding) { return sharding.type() == xla::OpSharding::MAXIMAL && sharding.tile_assignment_devices(0) == core_id; } // Returns the index of the return value of region in // `tf_device.parallel_execute` that represents cluster func output at
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
// logical device id. TPUExecute op for different logical device may have // different outputs depending on the output sharding configuration. mlir::LogicalResult GetOutputTypesForLogicalDeviceComputation( int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, mlir::tf_device::ClusterFuncOp cluster_func, llvm::SmallVectorImpl<mlir::Type>* output_types, llvm::SmallVectorImpl<int>* cluster_to_core_index);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:18:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
} replicate->setAttr(kDevicesAttr, builder->getDictionaryAttr(device_attrs)); } // Creates a `tf.TPUExecute` op that executes TPU program. LogicalResult BuildExecuteOp( const int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, llvm::ArrayRef<Value> inputs, tf_device::ClusterFuncOp cluster_func, OpBuilder* builder, TF::TPUExecuteOp* execute_op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
for (auto [core_id, result] : llvm::enumerate(partitioned_output->getResults())) { // outputs from replicate op are interleaved: // [(replica:0,core:0), (replica:1,core:0), ..., // (replica:0,core:1), (replica:1,core:1), ...] uint64_t output_id = core_id * num_replicas + replica_id + results.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
pkg/kubelet/cm/cpumanager/topology/topology.go
coreIDs = append(coreIDs, info.CoreID) } } } return cpuset.New(coreIDs...) } // CoresInSockets returns all of the core IDs associated with the given socket // IDs in this CPUDetails. func (d CPUDetails) CoresInSockets(ids ...int) cpuset.CPUSet { var coreIDs []int for _, id := range ids { for _, info := range d { if info.SocketID == id { coreIDs = append(coreIDs, info.CoreID)
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed May 03 16:26:09 UTC 2023 - 7.9K bytes - Viewed (0) -
pkg/kubelet/cm/cpumanager/topology/topology_test.go
45: {CoreID: 5, SocketID: 0, NUMANodeID: 0}, 46: {CoreID: 6, SocketID: 0, NUMANodeID: 0}, 47: {CoreID: 7, SocketID: 0, NUMANodeID: 0}, 48: {CoreID: 8, SocketID: 0, NUMANodeID: 0}, 49: {CoreID: 9, SocketID: 0, NUMANodeID: 0}, 50: {CoreID: 10, SocketID: 0, NUMANodeID: 1}, 51: {CoreID: 11, SocketID: 0, NUMANodeID: 1}, 52: {CoreID: 12, SocketID: 0, NUMANodeID: 1},
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed May 03 16:26:09 UTC 2023 - 31.7K bytes - Viewed (0) -
pkg/kubelet/cm/cpumanager/cpu_manager_test.go
NumSockets: 2, NumCores: 4, CPUDetails: map[int]topology.CPUInfo{ 0: {CoreID: 0, SocketID: 0}, 1: {CoreID: 1, SocketID: 0}, 2: {CoreID: 2, SocketID: 0}, 3: {CoreID: 3, SocketID: 0}, 4: {CoreID: 0, SocketID: 1}, 5: {CoreID: 1, SocketID: 1}, 6: {CoreID: 2, SocketID: 1}, 7: {CoreID: 3, SocketID: 1}, }, }, 0, cpuset.New(), topologymanager.NewFakeManager(),
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri Oct 06 13:16:15 UTC 2023 - 42.9K bytes - Viewed (0) -
pom.xml
<name>CodeLibs Project</name> <url>https://www.codelibs.org/</url> </organization> <scm> <connection>scm:git:******@****.***:codelibs/corelib.git</connection> <url>scm:git:******@****.***:codelibs/corelib.git</url> <developerConnection>scm:git:******@****.***:codelibs/corelib.git</developerConnection> </scm> <parent> <groupId>org.sonatype.oss</groupId> <artifactId>oss-parent</artifactId> <version>9</version>
Registered: Wed Jun 12 12:50:12 UTC 2024 - Last Modified: Thu Mar 07 01:58:02 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/saved_model_utils.cc
}); if (child_node_iter == current_node->children().end()) { return absl::nullopt; } node_id = child_node_iter->node_id(); current_node = &nodes.Get(node_id); } return node_id; } gtl::FlatMap<StringPiece, const AttrValueMap*, StringPieceHasher> NodeToAttrMap( const tensorflow::GraphDef& graphdef) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 12 19:17:46 UTC 2023 - 24K bytes - Viewed (0)