Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 10 for arg_min (0.24 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/tests/tfl_legalize_hlo.mlir

      // CHECK:  %cst = arith.constant dense<2> : tensor<1xi32>
      // CHECK:  %4 = "tfl.reduce_min"(%arg0, %cst) <{keep_dims = false}> : (tensor<4x32x256xf32>, tensor<1xi32>) -> tensor<4x32xf32>
      // CHECK:  %5 = "tfl.arg_min"(%arg0, %cst) : (tensor<4x32x256xf32>, tensor<1xi32>) -> tensor<4x32xi32>
      // CHECK:  return %4, %5 : tensor<4x32xf32>, tensor<4x32xi32>
    }
    
    // CHECK-LABEL:   func @convert_argmin_i16
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 40.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir

        func.return %0 : tensor<2x5xi32>
      }
    
      // CHECK-LABEL: arg_min
      func.func @arg_min(%arg0: tensor<6xf64>) -> tensor<i32> {
        // CHECK-NOT: ArgMin
        %0 = mhlo.constant dense<0> : tensor<i32>
        %1 = "tf.ArgMin"(%arg0, %0) : (tensor<6xf64>, tensor<i32>) -> tensor<i32>
        func.return %1 : tensor<i32>
      }
    
      // CHECK-LABEL: non_max_suppression_v4
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 38.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/schema/schema_v3b.fbs

      TILE = 69,
      EXPAND_DIMS = 70,
      EQUAL = 71,
      NOT_EQUAL = 72,
      LOG = 73,
      SUM = 74,
      SQRT = 75,
      RSQRT = 76,
      SHAPE = 77,
      POW = 78,
      ARG_MIN = 79,
      FAKE_QUANT = 80,
      REDUCE_PROD = 81,
      REDUCE_MAX = 82,
      PACK = 83,
      LOGICAL_OR = 84,
      ONE_HOT = 85,
      LOGICAL_AND = 86,
      LOGICAL_NOT = 87,
      UNPACK = 88,
      REDUCE_MIN = 89,
      FLOOR_DIV = 90,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 14:28:27 UTC 2024
    - 30K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/schema/schema.fbs

      TILE = 69,
      EXPAND_DIMS = 70,
      EQUAL = 71,
      NOT_EQUAL = 72,
      LOG = 73,
      SUM = 74,
      SQRT = 75,
      RSQRT = 76,
      SHAPE = 77,
      POW = 78,
      ARG_MIN = 79,
      FAKE_QUANT = 80,
      REDUCE_PROD = 81,
      REDUCE_MAX = 82,
      PACK = 83,
      LOGICAL_OR = 84,
      ONE_HOT = 85,
      LOGICAL_AND = 86,
      LOGICAL_NOT = 87,
      UNPACK = 88,
      REDUCE_MIN = 89,
      FLOOR_DIV = 90,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc

          auto arg_it = info.stack_var_arg_to_size_arg.find(i);
          if (arg_it == info.stack_var_arg_to_size_arg.end()) continue;
          auto it = data_var_to_size_var.find(call.getOperand(i));
          if (it == data_var_to_size_var.end()) {
            call.emitOpError("unknown stack");
            return failure();
          }
          assert(arg_it->second == new_operands.size());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/tests/post-quantize.mlir

      %2 = "tfl.dequantize"(%1) : (tensor<128x16x!quant.uniform<u8:f32, 3.906250e-03>>) -> tensor<128x16xf32>
    // CHECK-NEXT:  %[[argmax:.*]] = "tfl.arg_max"(%[[softmax]], %[[cst]]) : (tensor<128x16xf32>, tensor<i32>) -> tensor<128xi32>
      %3 = "tfl.arg_max"(%2, %cst) : (tensor<128x16xf32>, tensor<i32>) -> tensor<128xi32>
    // CHECK-NEXT:  return %[[softmax]], %[[argmax]] : tensor<128x16xf32>, tensor<128xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 19.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

        for (int64_t i = 0; i < call.getNumOperands(); ++i) {
          auto arg_it = info.buffer_arg_to_size_arg.find(i);
          if (arg_it == info.buffer_arg_to_size_arg.end()) continue;
          auto it = buffer_to_size->find(call.getOperand(i));
          if (it == buffer_to_size->end()) {
            call.emitOpError("unknown tensor list.");
            return failure();
          }
          assert(arg_it->second == new_operands.size());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  8. cluster/addons/addon-manager/kube-addons.sh

    # LIMITATIONS
    # 1. Exit code is probably not always correct.
    # 2. There are no unittests.
    # 3. Will not work if the total length of paths to addons is greater than
    #    bash can handle. Probably it is not a problem: ARG_MAX=2097152 on GCE.
    
    # cosmetic improvements to be done
    # 1. Improve the log function; add timestamp, file name, etc.
    # 2. Logging doesn't work from files that print things out.
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Sun Oct 15 05:40:38 UTC 2023
    - 10.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/aot/tests/tfcompile_test.cc

          xla::PrintHloProfile(fn.hlo_profile_printer_data(), fn.profile_counters(),
                               /*clock_rate_ghz=*/1.0);
      VLOG(1) << "Original HLO profile string:\n" << hlo_profile_as_string;
    
      // Replace Arg_n with argn when the MLIR bridge is used.
    #if defined(ENABLE_MLIR_BRIDGE_TEST)
      RE2::GlobalReplace(&hlo_profile_as_string, "(Arg_)([0-9].)", "arg\\2");
    #endif
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 26.4K bytes
    - Viewed (0)
  10. doc/asm.html

    and offsets match.
    On 32-bit systems, the low and high 32 bits of a 64-bit value are distinguished by adding
    a <code>_lo</code> or <code>_hi</code> suffix to the name, as in <code>arg_lo+0(FP)</code> or <code>arg_hi+4(FP)</code>.
    If a Go prototype does not name its result, the expected assembly name is <code>ret</code>.
    </p>
    
    <p>
    The <code>SP</code> pseudo-register is a virtual stack pointer
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Tue Nov 28 19:15:27 UTC 2023
    - 36.3K bytes
    - Viewed (1)
Back to top