Home
last modified time | relevance | path

Searched defs:MaxOp (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorModeKernel.cuh358 struct MaxOp { in compute_mode() struct
359 inline __device__ ModeUnsignedPair combine(ModeUnsignedPair a, ModeUnsignedPair b) const { in compute_mode()
363 inline __device__ ModeUnsignedPair warp_shfl_down(ModeUnsignedPair acc, int offset) const { in compute_mode()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_reduce_op.cc145 struct MaxOp { struct
147 …n(OpKernelContext *ctx, typename TTypes<T>::Scalar &s, const typename TTypes<T>::UnalignedVec &v) { in Run()
150 static StringPiece Name() { in Name()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
H A Dutils.h91 struct MaxOp { struct
96 struct MaxOp<float> { argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dreduction_ops.cc84 class MaxOp : public XlaReductionOp { class
86 explicit MaxOp(OpKernelConstruction* ctx) in MaxOp() function in tensorflow::__anonff6dd4aa0111::MaxOp
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dreduce.h285 struct MaxOp { struct
286 inline T operator()(const T& a) const { return a; } in operator()
287 inline T operator()(const T& a, const T& b) const { return (a > b) ? a : b; } in operator()
/aosp_15_r20/system/core/fs_mgr/libsnapshot/libsnapshot_cow/
H A Dtest_v3.cpp94 TEST_F(CowTestV3, MaxOp) { in TEST_F() argument
/aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/tools/codegen/core/
Dgen_huffman_decompressor.cc811 uint64_t MaxOp() const { in MaxOp() function
/aosp_15_r20/external/grpc-grpc/tools/codegen/core/
H A Dgen_huffman_decompressor.cc859 uint64_t MaxOp() const { in MaxOp() function
/aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/PowerPC/
H A DPPCInstrInfo.cpp3416 unsigned MaxOp = std::max(Op1, Op2); in swapMIOperands() local
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/PowerPC/
H A DPPCInstrInfo.cpp4412 unsigned MaxOp = std::max(Op1, Op2); in swapMIOperands() local