- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getCalledComputations (0.18 sec)
-
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) -
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)