- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for group_key (0.64 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-collective.mlir
} // ----- // CHECK: module attributes // CHECK-SAME{LITERAL}: tf2xla.collective_info.group_key = 1 // CHECK-SAME{LITERAL}: tf2xla.collective_info.group_size = 2 // CHECK-LABEL: func @collective_reduce_v2 func.func @collective_reduce_v2(%input: tensor<f32>) -> tensor<f32> { %group_key = "tf.Const"() { value = dense<1> : tensor<i32> } : () -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
StringRef(kGroupKeyAttrName.data(), kGroupKeyAttrName.size()), group_key, op, module))) { return failure(); } return success(); } LogicalResult SetCollectiveInfo(OpBuilder& builder, DenseIntElementsAttr replica_groups, Operation* op) { // Use special group_key 0 to represent "all available devices". This
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
<< " is required for CollectiveInfo but not found."; int32_t group_key = group_key_attr.getInt(); int32_t group_size = group_size_attr.getInt(); VLOG(2) << "Populating CollectiveInfo: group_key=" << group_key << " group_size=" << group_size; compilation_result->collective_info = {group_key, group_size, 0}; return absl::OkStatus(); } Status PopulateResultIOInfo(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td
Pattern< (TF_CollectiveReduceV2Op:$src_op $input, $group_size, $group_key, $instance_key, $ordering_token, $merge_op, $final_op, $communication_hint, $timeout_seconds, $is_stateless, $max_subdivs_per_device ), [(TF_CollectiveReduceV2Op $input, $group_size, $group_key, $instance_key, (EmptyList), $merge_op, $final_op, $communication_hint, $timeout_seconds,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 20.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
/// $group_assignment CompileTimeConstantOperand<TF::CollectiveAssignGroupV2Op, 0>, // $source_target_pairs CompileTimeConstantOperand<TF::CollectivePermuteOp, 1>, // $group_size, $group_key CompileTimeConstantOperand<TF::CollectiveReduceV2Op, 1, 2>, CompileTimeConstantOperand<TF::ConcatV2Op, -1>, // (variadic) $axis // $filter_sizes
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir
%group_size: tensor<i32>, %group_key: tensor<i32>, %instance_key: tensor<i32>, %token: tensor<!tf_type.resource<tensor<f32>>> ) -> tensor<3xi64> { %0 = "tf_device.cluster"() ({ // arg4 shall have been removed. // CHECK: tf.CollectiveReduceV2 // CHECK-NOT: [[TOKEN]] // CHECK-SAME: merge_op %0 = "tf.CollectiveReduceV2"(%input, %group_size, %group_key, %instance_key, %token) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 51.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
TF_DeleteStatus(status); TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(in)); TFE_OpSetAttrInt(op, "group_size", group_size); TFE_OpSetAttrInt(op, "group_key", 123); TFE_OpSetAttrInt(op, "instance_key", 456); TFE_OpSetAttrString(op, "merge_op", "Add", 3); TFE_OpSetAttrString(op, "final_op", "Id", 2); std::vector<int64_t> subdiv_offsets;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
docs/metrics/prometheus/alerts.md
"summary": "Instance 127.0.0.1:9000 has lot quorum on pool 0 on set 0" }, "externalURL": "http://fedora-minio:9093", "version": "4", "groupKey": "{}:{alertname=\"MinIOClusterTolerance\"}", "truncatedAlerts": 0 } ```
Registered: Sun Jun 16 00:44:34 UTC 2024 - Last Modified: Sun Jan 28 20:53:59 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); let results = (outs TF_Int32Tensor:$group_size, TF_Int32Tensor:$group_key ); } def TF_CollectiveBcastRecvOp : TF_Op<"CollectiveBcastRecv", []> { let summary = "Receives a tensor value broadcast from another device."; let arguments = (ins I64Attr:$group_size, I64Attr:$group_key, I64Attr:$instance_key, TF_ShapeAttr:$shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
staging/src/k8s.io/apiserver/pkg/endpoints/discovery/aggregated/handler.go
func (rdm *resourceDiscoveryManager) addGroupVersionLocked(source Source, groupName string, value apidiscoveryv2.APIVersionDiscovery) { if rdm.apiGroups == nil { rdm.apiGroups = make(map[groupKey]*apidiscoveryv2.APIGroupDiscovery) } key := groupKey{ source: source, name: groupName, } if existing, groupExists := rdm.apiGroups[key]; groupExists { // If this version already exists, replace it
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Sat Mar 02 00:29:39 UTC 2024 - 16.1K bytes - Viewed (0)