Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for FindOutsideCompiledOpsAtHead (0.25 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc 

    // computation or other ops that can be extracted, and have no operands from
// other ops in the TPU computation that cannot be extracted.
llvm::SmallVector<Operation*, 4> FindOutsideCompiledOpsAtHead(
    const mlir::TF::SideEffectAnalysis& side_effect_analysis,
    mlir::tf_device::ClusterOp cluster) {
  const auto& analysis = side_effect_analysis.GetAnalysisForFunc(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.6K bytes
    - Viewed (0)
Back to top