- Sort Score
- Result 10 results
- Languages All
Results 11 - 16 of 16 for ClusterFuncOp (0.38 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf_test.cc
/*is_in_fallback_enabled_mode=*/false)); FuncOp main = mlir_module_->lookupSymbol<mlir::func::FuncOp>("main"); ASSERT_TRUE(main); bool has_cluster_op = false; main.walk([&](mlir::tf_device::ClusterFuncOp cluster_op) { has_cluster_op = true; return WalkResult::advance(); }); EXPECT_TRUE(has_cluster_op); EXPECT_EQ(compilation_status.Delta(mlir::TF::kMlirPh1BridgeCounterReplicated,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc
return block_arg.getOwner()->getParentOp(); return value.getDefiningOp(); } void TPUAnnotateDynamicShapeInputsPass::runOnOperation() { getOperation().walk([&](tf_device::ClusterFuncOp cluster_func_op) { Builder builder(cluster_func_op->getContext()); // Skip non-tpu device cluster_func. auto cluster_id = cluster_func_op->getAttrOfType<StringAttr>(TF::kReplicationInfoAttr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
mlir::StringAttr::get(op->getContext(), hlosharding->ToString())); } mlir::LogicalResult ExtractInputsForLogicalDevices( const int num_cores_per_replica, mlir::tf_device::ClusterFuncOp cluster_func, mlir::OpBuilder* builder, llvm::SmallVectorImpl<llvm::SmallVector<mlir::Value, 4>>* input_list) { // Initialize the input list for each logical devices. input_list->reserve(num_cores_per_replica);
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/transforms/cluster_outlining.cc
builder->getStringAttr(kFuncAttr), mlir::SymbolRefAttr::get(builder->getContext(), outlined_func.getName())); builder->setInsertionPoint(cluster_op); auto cluster_func_op = builder->create<tf_device::ClusterFuncOp>( cluster_op.getLoc(), outlined_func.getFunctionType().getResults(), live_ins.getArrayRef(), cluster_op->getAttrs()); cluster_op.replaceAllUsesWith(cluster_func_op); cluster_op.erase(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
// Returns whether it's legal to inline a call to a function. bool isLegalToInline(Operation* call, Operation* callable, bool wouldBeCloned) const final { if (isa<ClusterFuncOp>(call)) return false; return true; } // Returns if its legal to inline 'src' region into the 'dest' region // attached to a TF Device operation.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0)