- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for arg_min (0.14 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.cc
#include "mlir/Support/LogicalResult.h" // from @llvm-project #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" namespace mlir { namespace odml { // Pattern matches the following reduction function for ArgMax/ArgMin coming // from PyTorch // %0 = compare{GT}(%lhs_value, %rhs_value) // %1 = select(%0, %lhs_value, %rhs_value) // %2 = compare{EQ}(%lhs_value, %rhs_value) // %3 = minimum(%lhs_index, %rhs_index)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 8K bytes - Viewed (0) -
src/math/dim_arm64.s
CMP R0, R2 BEQ isPosInf // normal case FMOVD R1, F0 FMOVD R2, F1 FMAXD F0, F1, F0 FMOVD F0, ret+16(FP) RET isPosInf: // return +Inf MOVD R0, ret+16(FP) RET // func archMin(x, y float64) float64 TEXT ·archMin(SB),NOSPLIT,$0 // -Inf special cases MOVD $NegInf, R0 MOVD x+0(FP), R1 CMP R0, R1 BEQ isNegInf MOVD y+8(FP), R2 CMP R0, R2 BEQ isNegInf // normal case FMOVD R1, F0
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Apr 15 15:48:19 UTC 2021 - 963 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.h
LogicalResult MatchReduceToArgMinMaxType2(mhlo::ReduceOp reduce_op, bool is_argmax); // Base class for converting mhlo::ReduceOp to TF/TFL ArgMax/ArgMin ops. template <typename Reduce, typename ArgReduce, typename BooleanReduce, bool is_argmax> class ConvertReduceOpToArgMinMax : public OpConversionPattern<mhlo::ReduceOp> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
src/cmd/vendor/golang.org/x/tools/go/analysis/passes/stdmethods/stdmethods.go
if !matchParams(pass, expect.args, args, "") || !matchParams(pass, expect.results, results, "") { expectFmt := id.Name + "(" + argjoin(expect.args) + ")" if len(expect.results) == 1 { expectFmt += " " + argjoin(expect.results) } else if len(expect.results) > 1 { expectFmt += " (" + argjoin(expect.results) + ")" } actual := typeString(sign) actual = strings.TrimPrefix(actual, "func")
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue May 09 01:28:01 UTC 2023 - 6.9K bytes - Viewed (0) -
src/math/dim_s390x.s
RET isMaxZero: MOVD $(1<<63), R4 // -0.0 CMPUBEQ R4, R8, +3(PC) MOVD R8, ret+16(FP) // return 0 RET MOVD R9, ret+16(FP) // return other 0 RET // func archMin(x, y float64) float64 TEXT ·archMin(SB),NOSPLIT,$0 // -Inf special cases MOVD $NegInf, R4 MOVD x+0(FP), R8 CMPUBEQ R4, R8, isNegInf MOVD y+8(FP), R9 CMPUBEQ R4, R9, isNegInf // NaN special cases
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Apr 15 15:48:19 UTC 2021 - 2K bytes - Viewed (0) -
src/math/dim_asm.go
// license that can be found in the LICENSE file. //go:build amd64 || arm64 || riscv64 || s390x package math const haveArchMax = true func archMax(x, y float64) float64 const haveArchMin = true
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Oct 28 18:17:57 UTC 2021 - 344 bytes - Viewed (0) -
src/math/dim_riscv64.s
// normal case FMAXD F0, F1, F0 MOVD F0, ret+16(FP) RET isMaxX: // return x MOVD F0, ret+16(FP) RET isMaxY: // return y MOVD F1, ret+16(FP) RET // func archMin(x, y float64) float64 TEXT ·archMin(SB),NOSPLIT,$0 MOVD x+0(FP), F0 MOVD y+8(FP), F1 FCLASSD F0, X5 FCLASSD F1, X6 // -Inf special cases MOV $NegInf, X7 BEQ X7, X5, isMinX BEQ X7, X6, isMinY
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Apr 15 15:48:19 UTC 2021 - 1.2K bytes - Viewed (0) -
src/math/dim_amd64.s
RET isMaxZero: MOVQ $(1<<63), AX // -0.0 CMPQ AX, R8 JEQ +3(PC) MOVQ R8, ret+16(FP) // return 0 RET MOVQ R9, ret+16(FP) // return other 0 RET // func archMin(x, y float64) float64 TEXT ·archMin(SB),NOSPLIT,$0 // -Inf special cases MOVQ $NegInf, AX MOVQ x+0(FP), R8 CMPQ AX, R8 JEQ isNegInf MOVQ y+8(FP), R9 CMPQ AX, R9 JEQ isNegInf
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Apr 15 15:48:19 UTC 2021 - 1.9K bytes - Viewed (0) -
src/math/dim_noasm.go
//go:build !amd64 && !arm64 && !riscv64 && !s390x package math const haveArchMax = false func archMax(x, y float64) float64 { panic("not implemented") } const haveArchMin = false func archMin(x, y float64) float64 { panic("not implemented")
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Oct 28 18:17:57 UTC 2021 - 410 bytes - Viewed (0) -
src/math/dim.go
// Min(-0, ±0) = Min(±0, -0) = -0 // // Note that this differs from the built-in function min when called // with NaN and -Inf. func Min(x, y float64) float64 { if haveArchMin { return archMin(x, y) } return min(x, y) } func min(x, y float64) float64 { // special cases switch { case IsInf(x, -1) || IsInf(y, -1): return Inf(-1) case IsNaN(x) || IsNaN(y): return NaN()
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Jun 15 19:45:12 UTC 2023 - 1.9K bytes - Viewed (0)