- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for arg_min (0.19 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/cc/experimental/libtf/object.h
int argument_index, TaggedValue& args_in, TArgsOut... args) { Handle h(std::move(args_in.tuple()[argument_index])); tensorflow::StatusOr<TSignatureArg> x = Cast<TSignatureArg>(std::move(h)); if (!x.ok()) return absl::InvalidArgumentError(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 11 08:05:36 UTC 2023 - 23.6K 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) -
src/cmd/vendor/golang.org/x/arch/arm/armasm/decode.go
arg_R_0 arg_R_12 arg_R_12_nzcv arg_R_16 arg_R_16_WB arg_R_8 arg_R_rotate arg_R_shift_R arg_R_shift_imm arg_SP arg_Sd arg_Sd_Dd arg_Dd_Sd arg_Sm arg_Sm_Dm arg_Sn arg_Sn_Dn arg_const arg_endian arg_fbits arg_fp_0 arg_imm24 arg_imm5 arg_imm5_32 arg_imm5_nz arg_imm_12at8_4at0 arg_imm_4at16_12at0 arg_imm_vfp arg_label24
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue Nov 22 17:16:14 UTC 2022 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantized_function_library_tf_drq.mlir
%r_min_without_zero = "tf.Min"(%input_1d, %dim) { keep_dims = true }: (tensor<?xf32>, tensor<1xi64>) -> tensor<1xf32> %r_min = "tf.Minimum"(%zero, %r_min_without_zero) : (tensor<1xf32>, tensor<1xf32>) -> tensor<1xf32> %r_max_f64 = "tf.Cast"(%r_max) : (tensor<1xf32>) -> tensor<1xf64> %r_min_f64 = "tf.Cast"(%r_min) : (tensor<1xf32>) -> tensor<1xf64> %i8_min = "tf.Const"() {value = dense<-128.0> : tensor<f32>} : () -> tensor<f32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 15:43:38 UTC 2023 - 12.2K 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)