- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for _skip_island_outlining (1.69 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_outline_island/executor_tpuv1_outline_tpu_island.mlir
// ----- // Test that islands inside a function with `_skip_island_outlining = true` are // skipped from outlining for both single-core and replicated case (i.e., the // `_tpu_v1_compat_outlined` module must be empty and no `PartitionedCallOp` is // created). Also check that `_skip_island_outlining` attribute is removed. // CHECK-LABEL: @func5 // CHECK-NOT: _skip_island_outlining // CHECK-NOT: PartitionedCallOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 04 03:54:58 UTC 2022 - 4.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h
// island outlining. The attribute is set in // `TpuV1BridgeExecutorIslandCoarsening` and removed in the subsequent // `TPUBridgeExecutorIslandOutlining` pass. inline constexpr llvm::StringRef kSkipIslandOutlining = "_skip_island_outlining"; // Function attribute to signal which argument contains bounded dynamic // dimension. inline constexpr llvm::StringRef kDynamicArgIndexAttr = "_dynamic_arg_index";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
getOperation().walk([&](IslandOp island_op) { auto parent_func = island_op->getParentOfType<func::FuncOp>(); auto skip_island_outlining = parent_func->getAttrOfType<BoolAttr>(mlir::TF::kSkipIslandOutlining); if (skip_island_outlining && skip_island_outlining.getValue()) { // Island was marked to be skipped. return WalkResult::advance(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0)