Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getCalledComputations (0.14 sec)

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

        if (!attrs)
          return op->emitError(
              "backend_config has no 'attributes' key or the attributes value is "
              "not a dictionary");
    
        auto calledComputations = op.getCalledComputations();
        if (!calledComputations || calledComputations.size() != 1)
          return op->emitError("expected exactly one called_computation");
    
        auto decomposition = mlir::cast<FlatSymbolRefAttr>(calledComputations[0]);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

        }
    
        auto called_computations = op.getCalledComputations();
        if (called_computations.size() != 1) {
          return op.emitOpError()
                 << "ApproxTopK takes exactly 1 called_computation.";
        }
        mlir::func::FuncOp callee = module_op_->lookupSymbol<mlir::func::FuncOp>(
            mlir::cast<FlatSymbolRefAttr>(op.getCalledComputations()[0]));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
Back to top