- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 382 for get_operations (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/verify_no_outside_compilation_markers_pass.cc
bool HasXlaOutsideCompilationMarker(Operation& op) { return op.getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; } void VerifyNoOutsideCompilationMarkersPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (IsDeviceClusterOp(*op) && HasChildLaunchDeviceOp(*op)) { std::string launch_error =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 22 19:52:08 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_inline_device_ops.cc
std::tie(old_val, new_val) = it; old_val.replaceAllUsesExcept(new_val, &return_op); } cluster_op.erase(); } void XlaInlineDeviceOpsPass::runOnOperation() { ModuleOp module = getOperation(); llvm::SmallVector<tf_device::ClusterOp, 4> ops; module.walk( [&](tf_device::ClusterOp cluster_op) { ops.push_back(cluster_op); }); for (auto cluster_op : ops) { InlineDeviceOp(cluster_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 00:59:46 UTC 2022 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_input_dialect_to_executor_pass.cc
std::string kClusterFuncOpName = "tf_device.cluster_func"; return op->getName().getStringRef().str() == kClusterFuncOpName; } void VerifyInputDialectToExecutorPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (!tensorflow::tf2xla::internal::IsInBridgeAcceptableDialects(op)) { std::string error = "op is in dialect " +
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 08 16:32:56 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tfxla_device_specific_transforms.cc
get_alg_op->replaceAllUsesWith(tf_const); get_alg_op->erase(); return success(); } void TFXLADeviceSpecificTransforms::runOnOperation() { if (!device_type_.hasValue()) return; auto func_op = getOperation(); auto walk_result = func_op->walk([&](TF::StatelessRandomGetAlgOp op) { if (failed(ConvertGetAlgOp(op))) { op->emitOpError( "Could not convert and remove Device specific information");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_splits.cc
} return predecessors; } struct TPUColocateSplits : public impl::TPUColocateSplitsPassBase<TPUColocateSplits> { void runOnOperation() override; }; void TPUColocateSplits::runOnOperation() { getOperation().walk([&](Operation* op) { if (auto split = llvm::dyn_cast<TF::SplitOp>(op)) { if (HasDevice(split) || split->getAttrOfType<ArrayAttr>(kClassAttr)) return WalkResult::advance();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 18:44:34 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_to_legacy_compile_and_replicate_attributes.cc
} return mlir::WalkResult::advance(); }); return failure(result.wasInterrupted()); } void ConvertToLegacyCompileAndReplicateAttributesPass::runOnOperation() { func::FuncOp func_op = getOperation(); if (failed(ConvertToLegacyAttributes(func_op))) return signalPassFailure(); } } // namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateConvertToLegacyCompileAndReplicateAttributesPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_clustering_pass.cc
class VerifyClusteringPass : public impl::VerifyClusteringPassBase<VerifyClusteringPass> { public: void runOnOperation() override; }; void VerifyClusteringPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (!tensorflow::tf2xla::internal::IsInBridgeAcceptableDialects(op)) { std::string error = "op is in dialect " +
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 20:33:37 UTC 2023 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/restore_function_name.cc
} call_op->setAttr(TF::kStablehloEntryFunctionAttrName, FlatSymbolRefAttr::get(original_function_name)); } void RestoreFunctionNamePass::runOnOperation() { ModuleOp module_op = getOperation(); MLIRContext* ctx = module_op.getContext(); OpBuilder builder(ctx); SymbolTable symbol_table(module_op); // TODO - b/298966126: Improve this logic if needed.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 08:32:43 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/import_quant_stats_pass.cc
auto stats_op = b.create<quantfork::StatisticsOp>( b.getUnknownLoc(), res, layer_stats, axis_stats, axis); res.replaceAllUsesWith(stats_op); stats_op.getOperation()->replaceUsesOfWith(stats_op, res); } void ImportQuantStatsPass::ImportAsStatsOps(OpBuilder b, Operation *op, int index,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc
// (x - mean) * scale / sqrt(variance + epsilon) + offset, // is then to compute (x * multiplier) + (offset - mean * multiplier). auto epsilon = materializeEpsilon( bn_op.getOperation(), bn_op.getEpsilonAttr(), fp_type, bn_op.getVariance(), variance_type, rewriter); if (!epsilon) { return failure(); } // Compute multiplier = scale / sqrt(variance + epsilon)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.2K bytes - Viewed (0)