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