Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for OperationPass (0.15 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

    constexpr StringRef kHybridFuncSuffix = "_hybrid_fn";
    
    class QuantizeCompositeFunctionsPass
        : public mlir::PassWrapper<QuantizeCompositeFunctionsPass,
                                   OperationPass<ModuleOp>> {
     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(QuantizeCompositeFunctionsPass)
    
      explicit QuantizeCompositeFunctionsPass() = default;
    
      explicit QuantizeCompositeFunctionsPass(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

      (void)applyPatternsAndFoldGreedily(func, std::move(phase_2_patterns));
    }
    
    }  // namespace
    
    // Creates an instance of the TensorFlow Lite dialect PrepareTF pass.
    std::unique_ptr<OperationPass<func::FuncOp>> CreatePrepareTFPass(
        bool unfold_batch_matmul, bool allow_bf16_and_f16_type_legalization,
        bool use_fake_quant_num_bits) {
      return std::make_unique<PrepareTFPass>(unfold_batch_matmul,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

        return signalPassFailure();
      }
    }
    
    }  // namespace
    
    namespace TFDevice {
    std::unique_ptr<OperationPass<ModuleOp>> CreateResourceOpLiftingPass() {
      return std::make_unique<ResourceOpLiftingPass>();
    }
    
    std::unique_ptr<OperationPass<ModuleOp>>
    CreateResourceOpLiftingForMainFunctionPass() {
      return std::make_unique<ResourceOpLiftingForMainFunctionPass>();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

    using mlir::BlockArgument;
    using mlir::DenseIntElementsAttr;
    using mlir::IRMapping;
    using mlir::Location;
    using mlir::LogicalResult;
    using mlir::ModuleOp;
    using mlir::OpBuilder;
    using mlir::Operation;
    using mlir::OperationPass;
    using mlir::OpOperand;
    using mlir::OpResult;
    using mlir::OwningOpRef;
    using mlir::RankedTensorType;
    using mlir::StringAttr;
    using mlir::StringRef;
    using mlir::SymbolTable;
    using mlir::Type;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

          auto _ = handler.ConsumeStatus();
        }
      }
    }
    
    }  // namespace
    
    /// Creates an instance of the TensorFlow Lite dialect LowerStaticTensorList
    /// pass.
    std::unique_ptr<OperationPass<ModuleOp>> TFL::CreateLowerStaticTensorListPass(
        bool allow_tensorlist_pass_through, bool default_to_single_batch,
        bool enable_dynamic_update_slice) {
      return std::make_unique<LowerStaticTensorListPass>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc

        module_op.emitError()
            << "Failed to compose stablehlo uniform quantized types.";
        signalPassFailure();
      }
    }
    
    }  // namespace
    
    std::unique_ptr<OperationPass<ModuleOp>>
    CreateComposeUniformQuantizedTypePass() {
      return std::make_unique<ComposeUniformQuantizedTypePass>();
    }
    
    static PassRegistration<ComposeUniformQuantizedTypePass> pass;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

      orig_while_op.body_function().erase();
      orig_while_op.erase();
    
      LOG(INFO) << "EmbeddingPipeliningPass::runOnOperation done.";
    }
    }  // namespace
    
    std::unique_ptr<OperationPass<mlir::ModuleOp>> CreateEmbeddingPipeliningPass() {
      return std::make_unique<EmbeddingPipeliningPass>();
    }
    
    }  // namespace TFDevice
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
Back to top