Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for arginfo0 (0.12 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/xla_call_module_deserialization.mlir

        // CHECK-NOT:    function_list
        // CHECK-SAME:   module = ""
        // CHECK-SAME:   _entry_function = @main_0,
    
        // `module` is stablehlo bytecode for:
        //  func.func @main(%arg0: tensor<?xi32> {jax.arg_info = "x", mhlo.sharding = "{replicated}"}, %arg1: tensor<*xi32>) -> (tensor<?xi32> {jax.result_info = ""}) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 02 18:38:51 UTC 2023
    - 6.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/xla_call_module_round_trip.mlir

      // CHECK-LABEL: func private @main_0
      // CHECK-SAME:    %[[ARG0:.*]]: tensor<?xi32> {jax.arg_info = "x", mhlo.sharding = "{replicated}"}
      // CHECK-SAME:    %[[ARG1:.*]]: tensor<*xi32>)
      // CHECK-SAME:    (tensor<?xi32> {jax.result_info = ""})
      // CHECK-SAME:    attributes {_from_xla_call_module}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 02 18:38:51 UTC 2023
    - 2.8K bytes
    - Viewed (0)
Back to top