- Sort Score
- Result 10 results
- Languages All
Results 91 - 100 of 168 for created (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
return a->isBeforeInBlock(b); }); // Create tf_device::ClusterOp before the last operation in the block that // is a part of a match set. auto back = cluster.operations.back(); auto loc = back->getLoc(); OpBuilder builder(back); auto cluster_op = builder.create<tf_device::ClusterOp>(loc, return_types, policy); // Create block in cluster_op's region and move 'cluster.operations' into
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 27.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
Value tensor_i32 = builder.create<TF::CastOp>( loc, tensor_type.clone(builder.getIntegerType(32)), tensor); auto reduced = builder.create<TF::SumOp>(loc, tensor_i32, reduction_indices_value, /*keep_dims=*/builder.getBoolAttr(true)); auto mul_op = builder.create<TF::MulOp>(loc, zp, reduced);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
// Add i32 -> i8 requantization. UniformQuantizeOp uniform_quant_op = rewriter.create<UniformQuantizeOp>( op.getLoc(), func_result_type, op.getResults()); cast<func::ReturnOp>(entry_func_op.getBody().front().getTerminator()) .setOperand(0, uniform_quant_op); } template <typename GemmStyleOp> // Creates a quantized bias pattern for static and dynamic shape case
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc
// be created). static void DumpModule(mlir::ModuleOp module, std::string file_prefix) { std::string prefix = GetDumpDirFromEnvVar(); if (prefix.empty()) return; auto* env = tensorflow::Env::Default(); auto status = env->RecursivelyCreateDir(prefix); if (!status.ok()) { LOG(WARNING) << "cannot create directory '" << prefix << "': " << status.message();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 18.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
// It's here so xla::Parameters that are created form block.getArguments will // have the proper shapes. TF_RETURN_IF_ERROR(RefineShapes(arg_shapes, module_op)); mlir::func::FuncOp main = module_op.lookupSymbol<mlir::func::FuncOp>("main"); mlir::Block& block = main.getRegion().front(); xla::XlaBuilder builder("main"); // Create xla_params. std::vector<xla::XlaOp> xla_params;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize.cc
// be categorized as cusom ops despite each of them may require different // behaviors. In that case, these ops can be marked in the custom map and // treated separately in this pass. auto custom_op = llvm::dyn_cast_or_null<CustomOp>(op); if (!custom_op) return false; // Custom op which is marked in the custom op map is quantizable.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc
return; } auto new_func_type = builder.getFunctionType(new_input_types, new_output_types); func.setType(new_func_type); } } // namespace // Creates an instance of the TensorFlow Lite modify io nodes pass. std::unique_ptr<OperationPass<func::FuncOp>> CreateModifyIONodesPass( Type input_type, Type output_type) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
return WalkResult::advance(); }); if (result.wasInterrupted()) { return failure(); } return builder.getArrayAttr(function_list); } // Creates a pruned module containing the XlaCallModule's entry function and // other functions transitively called by the entry function. FailureOr<OwningOpRef<ModuleOp>> PruneStablehloModule(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
} else { builder.setInsertionPointToStart(&block); } TypeAttr type_attr = TypeAttr::get(new_type); auto quantize = builder.create<TFL::QuantizeOp>(value.getLoc(), new_type, value, type_attr); auto dequantize = builder.create<TFL::DequantizeOp>( value.getLoc(), expressed_type, quantize.getOutput()); value.replaceAllUsesWith(dequantize);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
nodes_[inst] = node; return absl::OkStatus(); } bool IsEntryFunctionArg(BlockArgument arg) { return arg.getParentRegion()->getParentOfType<FuncOp>().getName() == "main"; } // Creates argument nodes from Block argument. If a name is supplied, that // name will be used instead of generating a unique name. Status Exporter::AddArgumentNode(BlockArgument arg, unsigned index,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0)