- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 101 for REWRITES (0.1 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc
input.getLoc(), input, rewriter.getDenseI64ArrayAttr(permutation)); } // Defers the transpose of the left-hand side (LHS) to the right-hand side and // the result of a binary operation. In detail, this rewrites the // `op(transpose(%rhs), %lhs)` to `transpose(op(%rhs, transpose(%lhs)))`. The // LHS transpose permutation must be a NCHW->NHWC permutation. template <typename OpT>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
src/cmd/fix/main.go
"os" "path/filepath" "sort" "strings" "cmd/internal/telemetry" ) var ( fset = token.NewFileSet() exitCode = 0 ) var allowedRewrites = flag.String("r", "", "restrict the rewrites to this comma-separated list") var forceRewrites = flag.String("force", "", "force these fixes to run even if the code looks updated") var allowed, force map[string]bool var (
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue May 14 19:41:17 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/runtime_passes.h
#include <memory> #include "llvm/ADT/StringRef.h" #include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "mlir/Pass/Pass.h" // from @llvm-project namespace mlir { namespace TFTPU { // Creates a pass that rewrites `tf_device.launch_func` on TPUs into TPU runtime // ops. std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>> CreateTPURewritePass( llvm::StringRef module_name = llvm::StringRef());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 10 18:58:57 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.h
#include "tensorflow/core/framework/types.h" #include "tensorflow/core/lib/core/status.h" namespace tensorflow { // An OpKernel that compiles an op to an XLA computation and runs it. Unlike // XlaLaunch this doesn't rely on any rewrites of the graphdef - it will run a // vanilla TensorFlow op as long as the bridge supports it. class XlaCompileOnDemandOp : public OpKernel { public: explicit XlaCompileOnDemandOp(OpKernelConstruction* ctx)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_launch_func_to_tf_call.cc
namespace mlir { namespace TFDevice { namespace { #define GEN_PASS_DEF_CONVERTLAUNCHFUNCTOTFCALLPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" // Rewrites tf_device::LaunchFuncOp into TF::PartitionedCallOp. struct ConvertLaunchFuncToTFCallPass : public impl::ConvertLaunchFuncToTFCallPassBase< ConvertLaunchFuncToTFCallPass> { void runOnOperation() override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 21:08:09 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.cc
pm.addPass(mlir::createInlinerPass()); pm.addNestedPass<mlir::func::FuncOp>(mlir::createCanonicalizerPass()); pm.addPass(mlir::quant::CreateCastBf16OpsToF32Pass()); // Optimizes the graph via cleanups, merges, rewrites, constant folding, // and edge case handling where possible. pm.addNestedPass<mlir::func::FuncOp>( mlir::TF::CreateDropWhileShapeInvariantPass()); pm.addNestedPass<mlir::func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
FindUserOfType<TFL::QuantizeOp>(op) != nullptr); } }; // Rewrites `stablehlo.convolution` into fused `tfl.conv_2d`. // If available, fuse bias and activation adjacent to `stablehlo.convolution`. // This RewritePattern rewrites both the following into `tfl.conv_2d` op: // // StableHLO Quantizer output: // * input: per-tensor qi8
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.td
let dependentDialects = ["tf_device::TensorFlowDeviceDialect"]; } def XlaRewritePass : Pass<"tf-xla-rewrite", "mlir::ModuleOp"> { let summary = "Rewrites partition calls into Xla launch ops to make the attached function run on XLA."; let description = [{ This pass rewrites `tf.PartitionedCall` and `tf.StatefulPartitionedCall` operations with `_xla_compile_device_type` attribute in a
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/verify_quant_legalization.cc
// some generic types that are legal in MHLO. This pass legalizes TF types into // types that are legal in MHLO. For example, TF::Qint8Type is converted to i8. // Rewrites here should run before TF to MHLO op legalizations are run. #include <memory> #include "absl/log/log.h" #include "llvm/ADT/STLExtras.h" #include "llvm/Support/Casting.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/runtime_passes.td
include "mlir/Pass/PassBase.td" def TPURewritePass : Pass<"tf-tpu-rewrite", "mlir::ModuleOp"> { let summary = "Rewrites a `tf_device.cluster_func` on TPUs into TPU runtime operations."; let description = [{ This pass rewrites a `tf_device.cluster_func` operation into a sequence of `tf._TPUCompileMlir` and `tf.TPUExecute` operations. `tf._TPUCompileMlir` contains a MLIR module that is
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 10 18:58:57 UTC 2024 - 10.7K bytes - Viewed (0)