Home
last modified time | relevance | path

Searched refs:UpdateOp (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dscatter_functor_gpu.cu.cc31 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::ASSIGN); \
32 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::ADD); \
33 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::SUB); \
34 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MUL); \
35 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::DIV); \
36 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MIN); \
37 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MAX);
45 DEFINE_GPU_SPECS_OP(bool, int32, scatter_op::UpdateOp::ASSIGN);
46 DEFINE_GPU_SPECS_OP(bool, int64, scatter_op::UpdateOp::ASSIGN);
Dscatter_functor.h41 enum class UpdateOp { ASSIGN, ADD, SUB, MUL, DIV, MIN, MAX }; enum
45 template <scatter_op::UpdateOp Op>
48 struct Assign<scatter_op::UpdateOp::ASSIGN> {
59 struct Assign<scatter_op::UpdateOp::ADD> {
70 struct Assign<scatter_op::UpdateOp::SUB> {
81 struct Assign<scatter_op::UpdateOp::MUL> {
92 struct Assign<scatter_op::UpdateOp::DIV> {
103 struct Assign<scatter_op::UpdateOp::MIN> {
116 struct Assign<scatter_op::UpdateOp::MAX> {
128 template <scatter_op::UpdateOp Op>
[all …]
Dscatter_op_gpu.cu.cc32 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::ASSIGN); \
33 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::ADD); \
34 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::SUB); \
35 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MUL); \
36 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::DIV); \
37 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MIN); \
38 DEFINE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MAX);
Dscatter_nd_op_gpu.cu.cc32 template <typename T, scatter_nd_op::UpdateOp Op>
38 struct LeftUpdate<T, scatter_nd_op::UpdateOp::ASSIGN> {
45 struct LeftUpdate<T, scatter_nd_op::UpdateOp::ADD> {
52 struct LeftUpdate<T, scatter_nd_op::UpdateOp::SUB> {
62 struct LeftUpdate<std::complex<T>, scatter_nd_op::UpdateOp::ADD> {
72 struct LeftUpdate<std::complex<T>, scatter_nd_op::UpdateOp::SUB> {
75 LeftUpdate<std::complex<T>, scatter_nd_op::UpdateOp::ADD>()(out, -val); in operator ()()
81 template <typename T, typename Index, scatter_nd_op::UpdateOp op, int IXDIM>
111 template <typename T, typename Index, scatter_nd_op::UpdateOp op, int IXDIM>
165 DECLARE_GPU_SPECS_INDEX_OP(T, Index, scatter_nd_op::UpdateOp::ASSIGN); \
[all …]
Dscatter_functor_gpu.cu.h34 template <typename T, scatter_op::UpdateOp op>
38 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::ASSIGN> {
43 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::ADD> {
48 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::SUB> {
53 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::MUL> {
58 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::DIV> {
63 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::MIN> {
68 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::MAX> {
72 template <typename T, typename Index, scatter_op::UpdateOp op>
92 template <typename T, typename Index, scatter_op::UpdateOp op>
[all …]
Dscatter_functor.cc46 DECLARE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::ASSIGN); \
47 DECLARE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::ADD); \
48 DECLARE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::SUB); \
49 DECLARE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MUL); \
50 DECLARE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::DIV); \
51 DECLARE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MIN); \
52 DECLARE_GPU_SPECS_OP(T, Index, scatter_op::UpdateOp::MAX);
Dscatter_nd_op_cpu_impl.h51 scatter_nd_op::UpdateOp OP>
59 class UpdateExecutor<Input, Update, Output, scatter_nd_op::UpdateOp::ASSIGN> {
68 class UpdateExecutor<Input, Update, Output, scatter_nd_op::UpdateOp::ADD> {
77 class UpdateExecutor<Input, Update, Output, scatter_nd_op::UpdateOp::SUB> {
90 template <typename T, typename Index, scatter_nd_op::UpdateOp OP, int IXDIM>
156 REGISTER_SCATTER_ND_INDEX(type, scatter_nd_op::UpdateOp::ASSIGN);
159 REGISTER_SCATTER_ND_INDEX(type, scatter_nd_op::UpdateOp::ADD); \
160 REGISTER_SCATTER_ND_INDEX(type, scatter_nd_op::UpdateOp::SUB);
163 REGISTER_SCATTER_ND_INDEX(string, scatter_nd_op::UpdateOp::ADD);
174 template <typename T, typename Index, scatter_nd_op::UpdateOp OP, int IXDIM>
[all …]
Dscatter_op.cc72 template <typename Device, typename T, typename Index, scatter_op::UpdateOp op>
156 template <typename T, typename Index, scatter_op::UpdateOp op>
256 REGISTER_SCATTER_KERNEL(type, dev, "ScatterAdd", scatter_op::UpdateOp::ADD); \
257 REGISTER_SCATTER_KERNEL(type, dev, "ScatterDiv", scatter_op::UpdateOp::DIV); \
258 REGISTER_SCATTER_KERNEL(type, dev, "ScatterMul", scatter_op::UpdateOp::MUL); \
259 REGISTER_SCATTER_KERNEL(type, dev, "ScatterSub", scatter_op::UpdateOp::SUB);
262 REGISTER_SCATTER_KERNEL(type, dev, "ScatterMin", scatter_op::UpdateOp::MIN); \
263 REGISTER_SCATTER_KERNEL(type, dev, "ScatterMax", scatter_op::UpdateOp::MAX);
267 scatter_op::UpdateOp::ASSIGN);
Dscatter_nd_op.cc132 c, functor::DoScatterNd<Device, T, Index, scatter_nd_op::UpdateOp::ADD>( in Compute()
139 scatter_nd_op::UpdateOp op>
225 scatter_nd_op::UpdateOp op>
350 scatter_nd_op::UpdateOp::ADD); \
352 scatter_nd_op::UpdateOp::ADD); \
354 scatter_nd_op::UpdateOp::SUB); \
356 type, dev, "ResourceScatterNdAdd", scatter_nd_op::UpdateOp::ADD); \
358 type, dev, "ResourceScatterNdSub", scatter_nd_op::UpdateOp::SUB);
365 scatter_nd_op::UpdateOp::ASSIGN); \
367 type, dev, "ResourceScatterNdUpdate", scatter_nd_op::UpdateOp::ASSIGN);
[all …]
Dscatter_nd_op.h40 enum class UpdateOp { ASSIGN, ADD, SUB }; enum
48 scatter_nd_op::UpdateOp op, int IXDIM>
67 scatter_nd_op::UpdateOp Op>
Dresource_variable_ops.cc753 template <typename Device, typename T, typename Index, scatter_op::UpdateOp op>
835 scatter_op::UpdateOp::ADD); \
837 scatter_op::UpdateOp::SUB); \
839 scatter_op::UpdateOp::MUL); \
841 scatter_op::UpdateOp::DIV); \
843 scatter_op::UpdateOp::ASSIGN);
846 scatter_op::UpdateOp::MIN); \
848 scatter_op::UpdateOp::MAX);
859 scatter_op::UpdateOp::ASSIGN);
861 scatter_op::UpdateOp::ASSIGN);
[all …]
Dsparse_tensor_dense_add_op.cc97 scatter_op::UpdateOp::ADD>()( \ in Compute()
125 struct ScatterNdFunctor<CPUDevice, T, Index, NDIMS, scatter_op::UpdateOp::ADD> {
Dsparse_tensor_dense_add_op.h31 scatter_op::UpdateOp op>
/external/swiftshader/third_party/LLVM/lib/Target/X86/
DX86CodeEmitter.cpp592 static const MCInstrDesc *UpdateOp(MachineInstr &MI, const X86InstrInfo *II, in UpdateOp() function
606 case X86::ADD16rr_DB: Desc = UpdateOp(MI, II, X86::OR16rr); break; in emitInstruction()
607 case X86::ADD32rr_DB: Desc = UpdateOp(MI, II, X86::OR32rr); break; in emitInstruction()
608 case X86::ADD64rr_DB: Desc = UpdateOp(MI, II, X86::OR64rr); break; in emitInstruction()
609 case X86::ADD16ri_DB: Desc = UpdateOp(MI, II, X86::OR16ri); break; in emitInstruction()
610 case X86::ADD32ri_DB: Desc = UpdateOp(MI, II, X86::OR32ri); break; in emitInstruction()
611 case X86::ADD64ri32_DB: Desc = UpdateOp(MI, II, X86::OR64ri32); break; in emitInstruction()
612 case X86::ADD16ri8_DB: Desc = UpdateOp(MI, II, X86::OR16ri8); break; in emitInstruction()
613 case X86::ADD32ri8_DB: Desc = UpdateOp(MI, II, X86::OR32ri8); break; in emitInstruction()
614 case X86::ADD64ri8_DB: Desc = UpdateOp(MI, II, X86::OR64ri8); break; in emitInstruction()
[all …]
/external/clang/lib/CodeGen/
DCGAtomic.cpp236 const llvm::function_ref<RValue(RValue)> &UpdateOp,
275 const llvm::function_ref<RValue(RValue)> &UpdateOp,
279 const llvm::function_ref<RValue(RValue)> &UpdateOp,
1493 const llvm::function_ref<RValue(RValue)> &UpdateOp, in EmitAtomicUpdateValue() argument
1535 RValue NewRVal = UpdateOp(UpRVal); in EmitAtomicUpdateValue()
1546 llvm::AtomicOrdering AO, const llvm::function_ref<RValue(RValue)> &UpdateOp, in EmitAtomicUpdateLibcall() argument
1565 EmitAtomicUpdateValue(CGF, *this, OldRVal, UpdateOp, DesiredAddr); in EmitAtomicUpdateLibcall()
1575 llvm::AtomicOrdering AO, const llvm::function_ref<RValue(RValue)> &UpdateOp, in EmitAtomicUpdateOp() argument
1597 EmitAtomicUpdateValue(CGF, *this, OldRVal, UpdateOp, NewAtomicAddr); in EmitAtomicUpdateOp()
1688 llvm::AtomicOrdering AO, const llvm::function_ref<RValue(RValue)> &UpdateOp, in EmitAtomicUpdate() argument
[all …]
DCodeGenFunction.h2637 const llvm::function_ref<RValue(RValue)> &UpdateOp,