- Sort Score
- Result 10 results
- Languages All
Results 51 - 58 of 58 for created (0.06 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
StringAttr callee_sym = cluster_func_op.getFuncAttr().getAttr(); MoveResourceArgsToEnd(symtab.lookup<func::FuncOp>(callee_sym)); } builder.setInsertionPoint(cluster_func_op); auto xla_launch_op = builder.create<TF::XlaLaunchOp>( cluster_func_op.getLoc(), cluster_func_op.getResultTypes(), /*constants=*/ValueRange({}), ValueRange(non_resource_args), ValueRange(resource_args), cluster_func_op.getFuncAttr());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/ConvertConst.cc
// original const and the qbarrier that led to the quantization. auto fusedLoc = rewriter.getFusedLoc( {qbarrier.getArg().getDefiningOp()->getLoc(), qbarrier.getLoc()}); auto newConstOp = rewriter.create<arith::ConstantOp>( fusedLoc, newConstValueType, cast<TypedAttr>(newConstValue)); rewriter.replaceOpWithNewOp<StorageCastOp>(qbarrier, qbarrier.getType(), newConstOp);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.cc
} } // namespace Status MapFunctionSignaturesFromTFSavedModelMLIR( mlir::ModuleOp module, llvm::function_ref<void(const TFRTSavedModelSignatureInfo&)> map_fn) { // Create bound inputs for each functions. mlir::SymbolTable symbol_table(module); tensorflow::Status status = absl::OkStatus(); module.walk([&symbol_table, map_fn, &status](mlir::func::FuncOp func) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/split_merged_operands.cc
func.emitError() << "Failed to duplicate values for the stateful op\n"; return signalPassFailure(); } } } } } // namespace /// Creates an instance of the TensorFlow Lite dialect SplitMergedOperands /// pass. std::unique_ptr<OperationPass<func::FuncOp>> CreateSplitMergedOperandsPass() { return std::make_unique<SplitMergedOperandsPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/pass_pipeline.cc
AddShapeLegalizationPasses(pm); pm.addPass(createReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass()); // ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass may create // duplicate constants. Add canonicalizer to deduplicate. pm.addNestedPass<func::FuncOp>(mlir::createCanonicalizerPass()); pm.addPass(TF::CreateXlaCallModuleSerializationPass()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_graph_optimization_pass.cc
#include "tensorflow/core/platform/status.h" #include "tensorflow/core/public/session_options.h" #include "tsl/platform/statusor.h" #define DEBUG_TYPE "run-tf-graph-optimization" namespace tensorflow { namespace { // Creates a pass to convert MLIR to Graph, run user-specified Graph // Optimization Passes and convert back to MLIR. // Constraints: This pass expects that all operations in the MLIR module either
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
if (auto block_arg = mlir::dyn_cast<BlockArgument>(read_var_op.getResource())) { if (block_arg.getOwner() != replicate_block) return; OpBuilder builder(shape_op); auto new_shape_op = builder.create<TF::VariableShapeOp>( shape_op.getLoc(), shape_op.getType(), replicate_op.GetReplicaOperandForBlockArgument(block_arg, /*replica=*/0));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
op->getOperands(), result_types, op->getAttrs()); for (int i = 0; i < op->getNumRegions(); ++i) { state.addRegion(); } Operation* new_op = builder.create(state); for (const auto& indexed_regions : llvm::enumerate(op->getRegions())) { Region& region = op->getRegion(indexed_regions.index()); IRMapping mapping; indexed_regions.value().cloneInto(®ion, mapping);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0)