Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for CreateDeviceIndexSelectorPass (0.42 sec)

  1. tensorflow/compiler/mlir/lite/tf_tfl_passes.cc

      standard_pipeline_options.form_clusters = pass_config.form_clusters;
      mlir::TF::CreateTFStandardPipeline(*pass_manager, standard_pipeline_options);
      pass_manager->addNestedPass<mlir::func::FuncOp>(
          mlir::TF::CreateDeviceIndexSelectorPass());
    
      // Add canonicalize pass to remove no-op session initializer pass.
      pass_manager->addPass(mlir::createCanonicalizerPass());
    
      if (pass_config.guarantee_all_funcs_one_use) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 18:45:51 UTC 2024
    - 25.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/passes.h

    std::unique_ptr<OperationPass<func::FuncOp>> CreateFusedKernelMatcherPass();
    
    // Creates function pass to select device index/fold tf.DeviceIndex.
    std::unique_ptr<OperationPass<func::FuncOp>> CreateDeviceIndexSelectorPass();
    
    // Creates function pass to replace InitializeTableFromTextFileV2Ops with
    // LookupTableImportV2Op ops.
    std::unique_ptr<OperationPass<func::FuncOp>> CreateInitTextFileToImportPass(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 31.8K bytes
    - Viewed (0)
Back to top