/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/ |
D | cutlassB_f16_notaligned_k128.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 128, 64, 128>::… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 128, 64, 128>::… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, false, false, 64, 64, 128>::k… in __launch_bounds__() argument 69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 64, 64, 128>::k… in __launch_bounds__() argument 88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 64, 64, 128>::k… in __launch_bounds__() argument
|
D | cutlassB_f16_notaligned_k65536.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 128, 64, 65536>… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 128, 64, 65536>… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, false, false, 64, 64, 65536>:… in __launch_bounds__() argument 69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 64, 64, 65536>:… in __launch_bounds__() argument 88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 64, 64, 65536>:… in __launch_bounds__() argument
|
D | cutlassB_f16_notaligned_k32.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, false, false, 64, 64, 32>::kN… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 64, 64, 32>::kN… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 64, 64, 32>::kN… in __launch_bounds__() argument
|
D | cutlassB_f16_notaligned_k64.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, false, false, 64, 64, 64>::kN… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 64, 64, 64>::kN… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 64, 64, 64>::kN… in __launch_bounds__() argument
|
D | cutlassB_f32_notaligned_k128.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm50, float, false, false, false, 64, 64, 128>::kNumThreads, in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm70, float, false, false, false, 64, 64, 128>::kNumThreads, in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm75, float, false, false, false, 64, 64, 128>::kNumThreads, in __launch_bounds__() argument
|
D | cutlassB_f32_notaligned_k65536.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm50, float, false, false, false, 64, 64, 65536>::kNumThrea… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm70, float, false, false, false, 64, 64, 65536>::kNumThrea… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm75, float, false, false, false, 64, 64, 65536>::kNumThrea… in __launch_bounds__() argument
|
D | cutlassB_f32_notaligned_k64.cu | 12 … AttentionBackwardKernel<cutlass::arch::Sm50, float, false, false, false, 64, 64, 64>::kNumThreads, in __launch_bounds__() argument 31 … AttentionBackwardKernel<cutlass::arch::Sm70, float, false, false, false, 64, 64, 64>::kNumThreads, in __launch_bounds__() argument 50 … AttentionBackwardKernel<cutlass::arch::Sm75, float, false, false, false, 64, 64, 64>::kNumThreads, in __launch_bounds__() argument
|
D | cutlassB_f32_notaligned_k32.cu | 12 … AttentionBackwardKernel<cutlass::arch::Sm50, float, false, false, false, 64, 64, 32>::kNumThreads, in __launch_bounds__() argument 31 … AttentionBackwardKernel<cutlass::arch::Sm70, float, false, false, false, 64, 64, 32>::kNumThreads, in __launch_bounds__() argument 50 … AttentionBackwardKernel<cutlass::arch::Sm75, float, false, false, false, 64, 64, 32>::kNumThreads, in __launch_bounds__() argument
|
D | cutlassB_f16_aligned_k65536.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 128, 64, 65536>:… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 128, 64, 65536>:… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, false, 128, 64, 65536>:… in __launch_bounds__() argument 69 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, false, false, 64, 64, 65536>::… in __launch_bounds__() argument 88 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 65536>::… in __launch_bounds__() argument 107 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 64, 64, 65536>::… in __launch_bounds__() argument 126 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, false, 64, 64, 65536>::… in __launch_bounds__() argument
|
D | cutlassB_f16_aligned_k128.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 128, 64, 128, tr… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 128, 128, 128, tr… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 128, 64, 128>::k… in __launch_bounds__() argument 69 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 128, 64, 128>::k… in __launch_bounds__() argument 88 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 128, 128, 128>::k… in __launch_bounds__() argument 107 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 128, tru… in __launch_bounds__() argument 126 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, false, 64, 64, 128, tru… in __launch_bounds__() argument 145 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, false, false, 64, 64, 128>::kN… in __launch_bounds__() argument 164 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 128>::kN… in __launch_bounds__() argument 183 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 64, 64, 128>::kN… in __launch_bounds__() argument [all …]
|
D | cutlassB_f16_notaligned_k65536_dropout.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, true, false, 128, 64, 65536>:… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, true, false, 128, 64, 65536>:… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, true, false, 64, 64, 65536>::… in __launch_bounds__() argument 69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, true, false, 64, 64, 65536>::… in __launch_bounds__() argument 88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, true, false, 64, 64, 65536>::… in __launch_bounds__() argument
|
D | cutlassB_f16_notaligned_k128_dropout.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, true, false, 128, 64, 128>::k… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, true, false, 128, 64, 128>::k… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, true, false, 64, 64, 128>::kN… in __launch_bounds__() argument 69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, true, false, 64, 64, 128>::kN… in __launch_bounds__() argument 88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, true, false, 64, 64, 128>::kN… in __launch_bounds__() argument
|
D | cutlassB_f32_aligned_k128.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm80, float, true, false, false, 128, 64, 128>::kNumThreads, in __launch_bounds__() argument 31 … AttentionBackwardKernel<cutlass::arch::Sm50, float, true, false, false, 64, 64, 128>::kNumThreads, in __launch_bounds__() argument 50 … AttentionBackwardKernel<cutlass::arch::Sm70, float, true, false, false, 64, 64, 128>::kNumThreads, in __launch_bounds__() argument 69 … AttentionBackwardKernel<cutlass::arch::Sm75, float, true, false, false, 64, 64, 128>::kNumThreads, in __launch_bounds__() argument 88 … AttentionBackwardKernel<cutlass::arch::Sm80, float, true, false, false, 64, 64, 128>::kNumThreads, in __launch_bounds__() argument
|
D | cutlassB_f32_aligned_k65536.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm80, float, true, false, false, 128, 64, 65536>::kNumThrea… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm50, float, true, false, false, 64, 64, 65536>::kNumThread… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm70, float, true, false, false, 64, 64, 65536>::kNumThread… in __launch_bounds__() argument 69 …AttentionBackwardKernel<cutlass::arch::Sm75, float, true, false, false, 64, 64, 65536>::kNumThread… in __launch_bounds__() argument 88 …AttentionBackwardKernel<cutlass::arch::Sm80, float, true, false, false, 64, 64, 65536>::kNumThread… in __launch_bounds__() argument
|
D | cutlassB_f32_aligned_k64.cu | 12 … AttentionBackwardKernel<cutlass::arch::Sm50, float, true, false, false, 64, 64, 64>::kNumThreads, in __launch_bounds__() argument 31 … AttentionBackwardKernel<cutlass::arch::Sm70, float, true, false, false, 64, 64, 64>::kNumThreads, in __launch_bounds__() argument 50 … AttentionBackwardKernel<cutlass::arch::Sm75, float, true, false, false, 64, 64, 64>::kNumThreads, in __launch_bounds__() argument 69 … AttentionBackwardKernel<cutlass::arch::Sm80, float, true, false, false, 64, 64, 64>::kNumThreads, in __launch_bounds__() argument
|
D | cutlassB_f32_aligned_k32.cu | 12 … AttentionBackwardKernel<cutlass::arch::Sm50, float, true, false, false, 64, 64, 32>::kNumThreads, in __launch_bounds__() argument 31 … AttentionBackwardKernel<cutlass::arch::Sm70, float, true, false, false, 64, 64, 32>::kNumThreads, in __launch_bounds__() argument 50 … AttentionBackwardKernel<cutlass::arch::Sm75, float, true, false, false, 64, 64, 32>::kNumThreads, in __launch_bounds__() argument 69 … AttentionBackwardKernel<cutlass::arch::Sm80, float, true, false, false, 64, 64, 32>::kNumThreads, in __launch_bounds__() argument
|
D | cutlassB_f16_aligned_k32.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 32, true… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 64, 64, 32, true>… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, false, false, 64, 64, 32>::kNu… in __launch_bounds__() argument 69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 32>::kNu… in __launch_bounds__() argument 88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 64, 64, 32>::kNu… in __launch_bounds__() argument 107 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 64, 64, 32>::kNum… in __launch_bounds__() argument
|
D | cutlassB_f16_aligned_k64.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 64, true… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 64, 64, 64, true>… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, false, false, 64, 64, 64>::kNu… in __launch_bounds__() argument 69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 64>::kNu… in __launch_bounds__() argument 88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 64, 64, 64>::kNu… in __launch_bounds__() argument 107 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 64, 64, 64>::kNum… in __launch_bounds__() argument
|
D | cutlassB_f32_notaligned_k65536_dropout.cu | 12 …AttentionBackwardKernel<cutlass::arch::Sm50, float, false, true, false, 64, 64, 65536>::kNumThread… in __launch_bounds__() argument 31 …AttentionBackwardKernel<cutlass::arch::Sm70, float, false, true, false, 64, 64, 65536>::kNumThread… in __launch_bounds__() argument 50 …AttentionBackwardKernel<cutlass::arch::Sm75, float, false, true, false, 64, 64, 65536>::kNumThread… in __launch_bounds__() argument
|
D | cutlassB_f32_notaligned_k64_dropout.cu | 12 … AttentionBackwardKernel<cutlass::arch::Sm50, float, false, true, false, 64, 64, 64>::kNumThreads, in __launch_bounds__() argument 31 … AttentionBackwardKernel<cutlass::arch::Sm70, float, false, true, false, 64, 64, 64>::kNumThreads, in __launch_bounds__() argument 50 … AttentionBackwardKernel<cutlass::arch::Sm75, float, false, true, false, 64, 64, 64>::kNumThreads, in __launch_bounds__() argument
|
D | cutlassB_f32_notaligned_k128_dropout.cu | 12 … AttentionBackwardKernel<cutlass::arch::Sm50, float, false, true, false, 64, 64, 128>::kNumThreads, in __launch_bounds__() argument 31 … AttentionBackwardKernel<cutlass::arch::Sm70, float, false, true, false, 64, 64, 128>::kNumThreads, in __launch_bounds__() argument 50 … AttentionBackwardKernel<cutlass::arch::Sm75, float, false, true, false, 64, 64, 128>::kNumThreads, in __launch_bounds__() argument
|
D | cutlassB_f32_notaligned_k32_dropout.cu | 12 … AttentionBackwardKernel<cutlass::arch::Sm50, float, false, true, false, 64, 64, 32>::kNumThreads, in __launch_bounds__() argument 31 … AttentionBackwardKernel<cutlass::arch::Sm70, float, false, true, false, 64, 64, 32>::kNumThreads, in __launch_bounds__() argument 50 … AttentionBackwardKernel<cutlass::arch::Sm75, float, false, true, false, 64, 64, 32>::kNumThreads, in __launch_bounds__() argument
|
/external/tensorflow/tensorflow/core/framework/ |
D | variant_encode_decode.h | 59 TypeResolver<T, false /* is_pod */, true /* Tensor */>, in EncodeVariantImpl() argument 68 TypeResolver<T, false /* is_pod */, false /* Tensor */, in EncodeVariantImpl() argument 76 void EncodeVariantImpl(const T& value, in EncodeVariantImpl() 77 TypeResolver<T, false /* is_pod */, false /* Tensor */, in EncodeVariantImpl() argument 85 bool DecodeVariantImpl(VariantTensorData data, in DecodeVariantImpl() 86 TypeResolver<T, true /* is_pod */, false /* Tensor */, in DecodeVariantImpl() argument 94 bool DecodeVariantImpl(VariantTensorData data, in DecodeVariantImpl() 95 TypeResolver<T, false /* is_pod */, true /* Tensor */, in DecodeVariantImpl() argument 105 TypeResolver<T, false /* is_pod */, false /* Tensor */, in DecodeVariantImpl() argument 115 bool DecodeVariantImpl(VariantTensorData data, in DecodeVariantImpl() [all …]
|
/external/grpc-grpc-java/core/src/test/java/io/grpc/internal/ |
D | MessageFramerTest.java | 83 verify(sink).deliverFrame(toWriteBuffer(new byte[] {0, 0, 0, 0, 2, 3, 14}), false, true, 1); in simplePayload() local 94 verify(sink).deliverFrame(toWriteBuffer(new byte[] {0, 0, 0, 0, 2}), false, false, 0); in simpleUnknownLengthPayload() local 95 verify(sink).deliverFrame(toWriteBuffer(new byte[] {3, 14}), false, true, 1); in simpleUnknownLengthPayload() local 109 toWriteBuffer(new byte[] {0, 0, 0, 0, 1, 3, 0, 0, 0, 0, 1, 14}), false, true, 2); in smallPayloadsShouldBeCombined() local 142 toWriteBuffer(new byte[] {0, 0, 0, 0, 8, 3, 14, 1, 5, 9, 2, 6}), false, false, 1); in payloadSplitBetweenSinks() local 146 verify(sink).deliverFrame(toWriteBuffer(new byte[] {5}), false, true, 0); in payloadSplitBetweenSinks() local 159 toWriteBuffer(new byte[] {0, 0, 0, 0, 3, 3, 14, 1, 0, 0, 0, 0}), false, false, 2); in frameHeaderSplitBetweenSinks() local 163 verify(sink).deliverFrame(toWriteBufferWithMinSize(new byte[] {1, 3}, 12), false, true, 0); in frameHeaderSplitBetweenSinks() local 173 verify(sink).deliverFrame(toWriteBuffer(new byte[] {0, 0, 0, 0, 0}), false, true, 1); in emptyPayloadYieldsFrame() local 183 verify(sink).deliverFrame(toWriteBuffer(new byte[] {0, 0, 0, 0, 0}), false, true, 1); in emptyUnknownLengthPayloadYieldsFrame() local [all …]
|
/external/jazzer-api/src/test/java/com/code_intelligence/jazzer/mutation/ |
D | ArgumentsMutatorTest.java | 57 false, in testStaticMethod() argument 87 false, in testStaticMethod() argument 119 false, in testStaticMethod() argument 163 false, in testInstanceMethod() argument 193 false, in testInstanceMethod() argument 225 false, in testInstanceMethod() argument 264 false, in testCrossOver() argument 279 false, in testCrossOver() argument
|