Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for PartitionedCallOp (0.11 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc

      }
      if (skip_resize_) {
        target.addLegalOp<TF::ResizeBilinearOp>();
        target.addLegalOp<TF::ResizeNearestNeighborOp>();
      }
      if (skip_partitioned_calls_) {
        target.addLegalOp<TF::PartitionedCallOp>();
        target.addLegalOp<TF::StatefulPartitionedCallOp>();
      }
    
      FrozenRewritePatternSet frozen_patterns(std::move(patterns));
      if (failed(applyPartialConversion(func, target, frozen_patterns))) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 7.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc

      auto function_uses =
          SymbolTable::getSymbolUses(func, &module.getBodyRegion());
      if (!function_uses.has_value()) return false;
      for (auto& function_use : function_uses.value()) {
        if (!llvm::isa<TF::PartitionedCallOp, TF::StatefulPartitionedCallOp>(
                function_use.getUser())) {
          return false;
        }
      }
      return true;
    }
    
    // Returns the `shared_name` attribute value if exists. If not, returns an
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.2K bytes
    - Viewed (0)
Back to top