Home
last modified time | relevance | path

Searched refs:scratch1 (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dbatch_norm_op.h83 typename TTypes<T>::Vec dg, typename TTypes<T>::Vec scratch1, in operator()
117 scratch1.device(d) = (var + var.constant(variance_epsilon)).rsqrt(); in operator()
127 out_backprop.reshape(rest_by_depth) * ((scratch1 * gamma) in operator()
131 dm.device(d) = -db * (scratch1 * gamma).eval(); in operator()
132 dg.device(d) = scratch2 * scratch1; in operator()
136 scratch1.reshape(one_by_depth).broadcast(rest_by_one); in operator()
137 dm.device(d) = -db * scratch1; in operator()
142 scratch1.device(d) = scratch1 * scratch1.constant(static_cast<T>(-0.5f)) / in operator()
146 dv.device(d) = scratch2 * (scratch1 * gamma).eval(); in operator()
148 dv.device(d) = scratch2 * scratch1; in operator()
Dfused_batch_norm_op.h70 typename TTypes<U>::Vec scratch1, in operator()
109 scratch1.device(d) = (pop_var + pop_var.constant(epsilon)).rsqrt(); in operator()
119 (y_backprop_rest_by_depth * ((scratch1 * scale) in operator()
124 scale_backprop.device(d) = scratch2 * scratch1; in operator()
Dbatch_norm_op.cc142 Tensor scratch1; in Compute() local
145 TensorShape({input.dim_size(3)}), &scratch1)); in Compute()
159 scratch1.vec<T>(), scratch2.vec<T>()); in Compute()
247 typename TTypes<T>::Vec scratch1, typename TTypes<T>::Vec scratch2); \
/external/libaom/libaom/aom_dsp/mips/
Dconvolve2_vert_dspr2.c33 uint32_t scratch1; in convolve_bi_vert_4_dspr2() local
100 [p2] "=&r"(p2), [scratch1] "=&r"(scratch1), [Temp1] "=&r"(Temp1), in convolve_bi_vert_4_dspr2()
123 uint32_t scratch1; in convolve_bi_vert_64_dspr2() local
190 [p2] "=&r"(p2), [scratch1] "=&r"(scratch1), [Temp1] "=&r"(Temp1), in convolve_bi_vert_64_dspr2()
Dconvolve8_vert_dspr2.c35 uint32_t scratch1, scratch2; in convolve_vert_4_dspr2() local
156 [n1] "=&r"(n1), [n2] "=&r"(n2), [scratch1] "=&r"(scratch1), in convolve_vert_4_dspr2()
183 uint32_t scratch1, scratch2; in convolve_vert_64_dspr2() local
305 [n1] "=&r"(n1), [n2] "=&r"(n2), [scratch1] "=&r"(scratch1), in convolve_vert_64_dspr2()
/external/libvpx/libvpx/vpx_dsp/mips/
Dconvolve2_vert_dspr2.c32 uint32_t scratch1; in convolve_bi_vert_4_dspr2() local
99 [p2] "=&r"(p2), [scratch1] "=&r"(scratch1), [Temp1] "=&r"(Temp1), in convolve_bi_vert_4_dspr2()
122 uint32_t scratch1; in convolve_bi_vert_64_dspr2() local
189 [p2] "=&r"(p2), [scratch1] "=&r"(scratch1), [Temp1] "=&r"(Temp1), in convolve_bi_vert_64_dspr2()
Dconvolve2_avg_dspr2.c32 uint32_t scratch1, scratch2; in convolve_bi_avg_vert_4_dspr2() local
106 [p2] "=&r"(p2), [scratch1] "=&r"(scratch1), in convolve_bi_avg_vert_4_dspr2()
131 uint32_t scratch1, scratch2; in convolve_bi_avg_vert_64_dspr2() local
206 [p2] "=&r"(p2), [scratch1] "=&r"(scratch1), in convolve_bi_avg_vert_64_dspr2()
Dconvolve8_vert_dspr2.c33 uint32_t scratch1, scratch2; in convolve_vert_4_dspr2() local
154 [n1] "=&r"(n1), [n2] "=&r"(n2), [scratch1] "=&r"(scratch1), in convolve_vert_4_dspr2()
181 uint32_t scratch1, scratch2; in convolve_vert_64_dspr2() local
303 [n1] "=&r"(n1), [n2] "=&r"(n2), [scratch1] "=&r"(scratch1), in convolve_vert_64_dspr2()
Dconvolve8_avg_dspr2.c34 uint32_t scratch1, scratch2; in convolve_avg_vert_4_dspr2() local
163 [n1] "=&r"(n1), [n2] "=&r"(n2), [scratch1] "=&r"(scratch1), in convolve_avg_vert_4_dspr2()
190 uint32_t scratch1, scratch2; in convolve_avg_vert_64_dspr2() local
320 [n1] "=&r"(n1), [n2] "=&r"(n2), [scratch1] "=&r"(scratch1), in convolve_avg_vert_64_dspr2()
/external/libvpx/libvpx/vp8/common/x86/
Dloopfilter_block_sse2_x86_64.asm18 movdqa scratch1, %2 ; v2
20 psubusb scratch1, %1 ; v2 - v1
22 por %1, scratch1 ; abs(v2 - v1)
72 movdqa scratch1, %1
73 psubsb scratch1, %4 ; signed_char_clamp(ps1 - qs1)
74 pandn scratch2, scratch1 ; vp8_filter &= hev
78 movdqa scratch1, %3
79 psubsb scratch1, %2 ; qs0 - ps0
80 paddsb scratch2, scratch1 ; vp8_filter += (qs0 - ps0)
81 paddsb scratch2, scratch1 ; vp8_filter += (qs0 - ps0)
[all …]
/external/u-boot/arch/arm/mach-omap2/am33xx/
Dboard.c302 writel(magic, &rtc->scratch1); in update_rtc_magic()
381 u32 scratch1; in rtc_only() local
384 scratch1 = readl(&rtc->scratch1); in rtc_only()
393 if ((scratch1 & 0xffff) != RTC_MAGIC_VAL) in rtc_only()
399 writel(0, &rtc->scratch1); in rtc_only()
406 rtc_only_update_board_type(scratch1 >> RTC_BOARD_TYPE_SHIFT); in rtc_only()
/external/llvm/test/CodeGen/AMDGPU/
Dscratch-buffer.ll20 %scratch1 = alloca [8192 x i32]
25 %scratchptr1 = getelementptr [8192 x i32], [8192 x i32]* %scratch1, i32 0, i32 0
37 %else_ptr = getelementptr [8192 x i32], [8192 x i32]* %scratch1, i32 0, i32 %else_offset
58 %scratch1 = alloca [8192 x i32]
66 %scratchptr1 = getelementptr [8192 x i32], [8192 x i32]* %scratch1, i32 0, i32 %offset1
78 %else_ptr = getelementptr [8192 x i32], [8192 x i32]* %scratch1, i32 0, i32 %else_offset
/external/protobuf/src/google/protobuf/util/
Dfield_comparator.cc98 string scratch1; in Compare() local
102 message_1, field, index_1, &scratch1), in Compare()
108 string scratch1; in Compare() local
112 reflection_1->GetStringReference(message_1, field, &scratch1), in Compare()
/external/swiftshader/third_party/llvm-7.0/llvm/test/CodeGen/AMDGPU/
Dscratch-buffer.ll19 %scratch1 = alloca [8192 x i32], addrspace(5)
24 %scratchptr1 = getelementptr [8192 x i32], [8192 x i32] addrspace(5)* %scratch1, i32 0, i32 0
36 …%else_ptr = getelementptr [8192 x i32], [8192 x i32] addrspace(5)* %scratch1, i32 0, i32 %else_off…
59 %scratch1 = alloca [8192 x i32], addrspace(5)
67 …%scratchptr1 = getelementptr [8192 x i32], [8192 x i32] addrspace(5)* %scratch1, i32 0, i32 %offse…
79 …%else_ptr = getelementptr [8192 x i32], [8192 x i32] addrspace(5)* %scratch1, i32 0, i32 %else_off…
Dmemory-legalizer-multiple-mem-operands-nontemporal-1.mir11 %scratch1 = alloca [8192 x i32], addrspace(5)
14 %scratchptr12 = bitcast [8192 x i32] addrspace(5)* %scratch1 to i32 addrspace(5)*
25 …%else_ptr = getelementptr [8192 x i32], [8192 x i32] addrspace(5)* %scratch1, i32 0, i32 %else_off…
107 - { id: 1, name: scratch1, type: default, offset: 32772, size: 32768,
Dmemory-legalizer-multiple-mem-operands-nontemporal-2.mir11 %scratch1 = alloca [8192 x i32], addrspace(5)
14 %scratchptr12 = bitcast [8192 x i32] addrspace(5)* %scratch1 to i32 addrspace(5)*
25 …%else_ptr = getelementptr [8192 x i32], [8192 x i32] addrspace(5)* %scratch1, i32 0, i32 %else_off…
107 - { id: 1, name: scratch1, type: default, offset: 32772, size: 32768,
/external/v8/src/builtins/arm64/
Dbuiltins-arm64.cc678 Register scratch1, Register scratch2, Register scratch3) { in ReplaceClosureCodeWithOptimizedCode() argument
681 __ Mov(scratch1, optimized_code); // Write barrier clobbers scratch1 below. in ReplaceClosureCodeWithOptimizedCode()
682 __ RecordWriteField(closure, JSFunction::kCodeOffset, scratch1, scratch2, in ReplaceClosureCodeWithOptimizedCode()
721 Register scratch1, Register scratch2, in MaybeTailCallOptimizedCodeSlot() argument
730 !AreAliased(feedback_vector, x0, x1, x3, scratch1, scratch2, scratch3)); in MaybeTailCallOptimizedCodeSlot()
735 Register optimized_code_entry = scratch1; in MaybeTailCallOptimizedCodeSlot()
823 Register bytecode, Register scratch1, in AdvanceBytecodeOffsetOrReturn() argument
825 Register bytecode_size_table = scratch1; in AdvanceBytecodeOffsetOrReturn()
867 __ Ldr(scratch1.W(), MemOperand(bytecode_size_table, bytecode, LSL, 2)); in AdvanceBytecodeOffsetOrReturn()
868 __ Add(bytecode_offset, bytecode_offset, scratch1); in AdvanceBytecodeOffsetOrReturn()
[all …]
/external/libopus/celt/
Dkiss_fft.c120 kiss_fft_cpx scratch0, scratch1; in kf_bfly4() local
124 C_ADD( scratch1 , Fout[1] , Fout[3] ); in kf_bfly4()
125 C_SUB( Fout[2], *Fout, scratch1 ); in kf_bfly4()
126 C_ADDTO( *Fout , scratch1 ); in kf_bfly4()
127 C_SUB( scratch1 , Fout[1] , Fout[3] ); in kf_bfly4()
129 Fout[1].r = ADD32_ovflw(scratch0.r, scratch1.i); in kf_bfly4()
130 Fout[1].i = SUB32_ovflw(scratch0.i, scratch1.r); in kf_bfly4()
131 Fout[3].r = SUB32_ovflw(scratch0.r, scratch1.i); in kf_bfly4()
132 Fout[3].i = ADD32_ovflw(scratch0.i, scratch1.r); in kf_bfly4()
/external/v8/src/mips/
Dmacro-assembler-mips.cc1441 Register scratch1, Register scratch2) { in AddPair() argument
1444 Addu(scratch1, left_low, right_low); in AddPair()
1445 Sltu(scratch3, scratch1, left_low); in AddPair()
1448 Move(dst_low, scratch1); in AddPair()
1454 Register scratch1, Register scratch2) { in SubPair() argument
1458 Subu(scratch1, left_low, right_low); in SubPair()
1461 Move(dst_low, scratch1); in SubPair()
1467 Register scratch1, Register scratch2) { in MulPair() argument
1470 Mulu(scratch2, scratch1, left_low, right_low); in MulPair()
1475 Move(dst_low, scratch1); in MulPair()
[all …]
/external/v8/src/builtins/ia32/
Dbuiltins-ia32.cc324 Register scratch1, Register scratch2, in Generate_StackOverflowCheck() argument
332 __ mov(scratch1, __ StaticVariable(real_stack_limit)); in Generate_StackOverflowCheck()
336 __ sub(scratch2, scratch1); in Generate_StackOverflowCheck()
339 __ mov(scratch1, num_args); in Generate_StackOverflowCheck()
341 __ add(scratch1, Immediate(1)); in Generate_StackOverflowCheck()
343 __ shl(scratch1, kPointerSizeLog2); in Generate_StackOverflowCheck()
345 __ cmp(scratch2, scratch1); in Generate_StackOverflowCheck()
427 Register scratch1) { in GetSharedFunctionInfoBytecode() argument
430 __ CmpObjectType(sfi_data, INTERPRETER_DATA_TYPE, scratch1); in GetSharedFunctionInfoBytecode()
572 Register scratch1, Register scratch2, Register scratch3) { in ReplaceClosureCodeWithOptimizedCode() argument
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc194 auto scratch1 = xla::Rsqrt(xla::Add(var, epsilon)); in Compile() local
206 xla::Mul(grad_backprop, xla::Mul(scratch1, scale), {feature_index}); in Compile()
207 scale_backprop = xla::Mul(scratch1, scratch2); in Compile()
/external/v8/src/mips64/
Dmacro-assembler-mips64.cc2082 Register scratch1 = t8; in Neg_s() local
2091 mfc1(scratch1, fs); in Neg_s()
2093 Xor(scratch1, scratch1, scratch2); in Neg_s()
2094 mtc1(scratch1, fd); in Neg_s()
2107 Register scratch1 = t8; in Neg_d() local
2116 dmfc1(scratch1, fs); in Neg_d()
2118 Xor(scratch1, scratch1, scratch2); in Neg_d()
2119 dmtc1(scratch1, fd); in Neg_d()
2263 Register scratch1 = temps.Acquire(); in Trunc_l_ud() local
2264 li(scratch1, 0x7FFFFFFFFFFFFFFF); in Trunc_l_ud()
[all …]
/external/v8/src/builtins/x64/
Dbuiltins-x64.cc486 Register scratch1) { in GetSharedFunctionInfoBytecode() argument
489 __ CmpObjectType(sfi_data, INTERPRETER_DATA_TYPE, scratch1); in GetSharedFunctionInfoBytecode()
634 Register scratch1, Register scratch2, Register scratch3) { in ReplaceClosureCodeWithOptimizedCode() argument
638 __ movp(scratch1, optimized_code); // Write barrier clobbers scratch1 below. in ReplaceClosureCodeWithOptimizedCode()
639 __ RecordWriteField(closure, JSFunction::kCodeOffset, scratch1, scratch2, in ReplaceClosureCodeWithOptimizedCode()
643 static void LeaveInterpreterFrame(MacroAssembler* masm, Register scratch1, in LeaveInterpreterFrame() argument
645 Register args_count = scratch1; in LeaveInterpreterFrame()
677 Register scratch1, Register scratch2, in MaybeTailCallOptimizedCodeSlot() argument
685 DCHECK(!AreAliased(feedback_vector, rax, rdx, rdi, scratch1, scratch2, in MaybeTailCallOptimizedCodeSlot()
691 Register optimized_code_entry = scratch1; in MaybeTailCallOptimizedCodeSlot()
[all …]
/external/v8/src/arm/
Dcode-stubs-arm.cc478 Register scratch1 = r5; in Generate() local
485 __ Move(scratch1, ExternalReference::isolate_address(masm->isolate())); in Generate()
486 __ push(scratch1); in Generate()
507 __ add(scratch1, scratch0, in Generate()
509 __ str(scratch1, MemOperand(r0, 1 * kPointerSize)); in Generate()
/external/u-boot/arch/arm/include/asm/
Ddavinci_rtc.h40 unsigned int scratch1; member

123