GraphXlaCallModule Method

public Operation XlaCallModule(
	Output args,
	long version,
	string module,
	long[][] Sout,
	DataType[] Tout,
	string[] dim_args_spec = null,
	string[] platforms = null,
	bool has_token_input_output = false,
	string[] disabled_checks = null,
	string opName = "XlaCallModule"
)

Public Function XlaCallModule ( 
	args As Output,
	version As Long,
	module As String,
	Sout As Long()(),
	Tout As DataType(),
	Optional dim_args_spec As String() = Nothing,
	Optional platforms As String() = Nothing,
	Optional has_token_input_output As Boolean = false,
	Optional disabled_checks As String() = Nothing,
	Optional opName As String = "XlaCallModule"
) As Operation

public:
Operation^ XlaCallModule(
	Output^ args, 
	long long version, 
	String^ module, 
	array<array<long long>^>^ Sout, 
	array<DataType>^ Tout, 
	array<String^>^ dim_args_spec = nullptr, 
	array<String^>^ platforms = nullptr, 
	bool has_token_input_output = false, 
	array<String^>^ disabled_checks = nullptr, 
	String^ opName = L"XlaCallModule"
)

member XlaCallModule : 
        args : Output * 
        version : int64 * 
        module : string * 
        Sout : int64[][] * 
        Tout : DataType[] * 
        ?dim_args_spec : string[] * 
        ?platforms : string[] * 
        ?has_token_input_output : bool * 
        ?disabled_checks : string[] * 
        ?opName : string 
(* Defaults:
        let _dim_args_spec = defaultArg dim_args_spec null
        let _platforms = defaultArg platforms null
        let _has_token_input_output = defaultArg has_token_input_output false
        let _disabled_checks = defaultArg disabled_checks null
        let _opName = defaultArg opName "XlaCallModule"
*)
-> Operation 

Parameters

args

Type: Emgu.TFOutput
Input to the operation: A list of `Tensor` with possibly different types to be passed as arguments to the `module`. These are the actual arguments and do not include the platform argument (see `platforms`) nor the dimension arguments (see `dim_args_spec`).

version

Type: SystemInt64
Tracks changes the semantics of the op, to support backwards compatibility. Minimum supported version is 2. From version 2, the op carries a StableHLO text or bytecode `module`. From version 3, the op also supports the `platforms` attribute. From version 4, the op carries a StableHLO module with compatibility guarantees. From version 5, XLACallModule can include `stablehlo.custom_call` op to execute tf functions. From version 6 the op supports the `disabled_checks` attribute. See more versioning details at https://github.com/search?q=repo%3Atensorflow%2Ftensorflow+path%3Axla_call_module+%22int+VERSION_MAXIMUM_SUPPORTED%22&type=code.

module

Type: SystemString
A serialized computation, a text or bytecode representation of an mlir.Module. The return type must be a tuple if and only if the `Sout` is a list with 0 or more than 1 elements. The length of `Tout` and `Sout` must match. This op always returns a tuple of results, even if the module returns a single result.

Sout

Type: SystemInt64
List of output tensor shapes.

Tout

Type: Emgu.TFDataType
List of output tensor data types.

dim_args_spec (Optional)

Type: SystemString
this attribute is not supported anymore.

platforms (Optional)

Type: SystemString
the list of platforms supported by `module`. The list can contain the strings "CPU", "CUDA", "ROCM", or "TPU". It is an error to compile this op for a platform that does not appear in the list. This check can be disabled using `disabled_checks`. If the list contains more than one platform, then the `module` takes one additional 0-dimensional integer-tensor parameter in the first position, encoding the index in `platforms` of the current compilation platform. This parameter has value 0 if the plaform is not among `platforms` and the check has been disabled. The list can be empty in old versions (earlier than 6) to denote that no platform checking must be performed at loading time.

has_token_input_output (Optional)

Type: SystemBoolean
If true, the embedded StableHLO module's main function must take a `!stablehlo.token` as its first argument and returns a token as its first result. This can be used in conjunction with the TF2XLA's side effect mechanism in order to model side effects. This is used only in versions prior to version 9. After that, the number and position of tokens among the arguments and results are obtained from the main function type. This allows us to support more than one token and not necessarily at the start.

disabled_checks (Optional)

Type: SystemString
A list of strings describing the safety checks that were disabled at serialization time. This attribute was added in version 6. For more details see https://github.com/search?q=repo%3Agoogle%2Fjax+path%3Ajax_export+%22class+DisabledSafetyCheck%22&type=code. This list, supplemented with a comma-separate list of directives specified using the flag --tf_xla_call_module_disabled_checks, is used at module loading time to skip the corresponding checks.

opName (Optional)

Type: SystemString
The name of the operation

Return Value

Reference