Home
last modified time | relevance | path

Searched refs:UpdateOp (Results 1 – 23 of 23) 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);
46 DEFINE_GPU_SPECS_OP(bool, int32, scatter_op::UpdateOp::ASSIGN);
47 DEFINE_GPU_SPECS_OP(bool, int64, scatter_op::UpdateOp::ASSIGN);
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> {
59 struct LeftUpdate<T, scatter_nd_op::UpdateOp::MAX> {
66 struct LeftUpdate<T, scatter_nd_op::UpdateOp::MIN> {
76 struct LeftUpdate<std::complex<T>, scatter_nd_op::UpdateOp::ADD> {
86 struct LeftUpdate<std::complex<T>, scatter_nd_op::UpdateOp::SUB> {
97 template <typename T, typename Index, scatter_nd_op::UpdateOp op, int IXDIM>
127 template <typename T, typename Index, scatter_nd_op::UpdateOp op, int IXDIM>
[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_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.h48 scatter_nd_op::UpdateOp OP>
57 scatter_nd_op::UpdateOp::ASSIGN> {
66 class UpdateExecutor<T, Input, Update, Output, scatter_nd_op::UpdateOp::ADD> {
75 class UpdateExecutor<T, Input, Update, Output, scatter_nd_op::UpdateOp::SUB> {
84 class UpdateExecutor<T, Input, Update, Output, scatter_nd_op::UpdateOp::MIN> {
93 class UpdateExecutor<T, Input, Update, Output, scatter_nd_op::UpdateOp::MAX> {
106 template <typename T, typename Index, scatter_nd_op::UpdateOp OP, int IXDIM>
173 REGISTER_SCATTER_ND_INDEX(type, scatter_nd_op::UpdateOp::ASSIGN);
176 REGISTER_SCATTER_ND_INDEX(type, scatter_nd_op::UpdateOp::ADD); \
177 REGISTER_SCATTER_ND_INDEX(type, scatter_nd_op::UpdateOp::SUB);
[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> {
50 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::SUB> {
57 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::MUL> {
64 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::DIV> {
71 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::MIN> {
78 struct ScatterOpKernelBody<T, scatter_op::UpdateOp::MAX> {
84 template <typename T, typename Index, scatter_op::UpdateOp op>
105 template <typename T, typename Index, scatter_op::UpdateOp op>
[all …]
Dscatter_functor.h39 enum class UpdateOp { ASSIGN, ADD, SUB, MUL, DIV, MIN, MAX }; enum
43 template <scatter_op::UpdateOp Op>
46 struct Assign<scatter_op::UpdateOp::ASSIGN> {
57 struct Assign<scatter_op::UpdateOp::ADD> {
68 struct Assign<scatter_op::UpdateOp::SUB> {
79 struct Assign<scatter_op::UpdateOp::MUL> {
90 struct Assign<scatter_op::UpdateOp::DIV> {
101 struct Assign<scatter_op::UpdateOp::MIN> {
114 struct Assign<scatter_op::UpdateOp::MAX> {
130 template <typename Device, typename T, typename Index, scatter_op::UpdateOp op>
[all …]
Dscatter_nd_op.cc129 c, functor::DoScatterNd<Device, T, Index, scatter_nd_op::UpdateOp::ADD>( in Compute()
136 scatter_nd_op::UpdateOp op>
230 scatter_nd_op::UpdateOp op>
421 scatter_nd_op::UpdateOp::ADD); \
423 scatter_nd_op::UpdateOp::ADD); \
425 scatter_nd_op::UpdateOp::SUB); \
427 type, dev, "ResourceScatterNdAdd", scatter_nd_op::UpdateOp::ADD); \
429 type, dev, "ResourceScatterNdSub", scatter_nd_op::UpdateOp::SUB);
433 "ScatterNdNonAliasingAdd", scatter_nd_op::UpdateOp::ADD); \
435 scatter_nd_op::UpdateOp::ADD); \
[all …]
Dscatter_op.cc66 template <typename Device, typename T, typename Index, scatter_op::UpdateOp op>
161 REGISTER_SCATTER_KERNEL(type, dev, "ScatterAdd", scatter_op::UpdateOp::ADD); \
162 REGISTER_SCATTER_KERNEL(type, dev, "ScatterDiv", scatter_op::UpdateOp::DIV); \
163 REGISTER_SCATTER_KERNEL(type, dev, "ScatterMul", scatter_op::UpdateOp::MUL); \
164 REGISTER_SCATTER_KERNEL(type, dev, "ScatterSub", scatter_op::UpdateOp::SUB);
167 REGISTER_SCATTER_KERNEL(type, dev, "ScatterMin", scatter_op::UpdateOp::MIN); \
168 REGISTER_SCATTER_KERNEL(type, dev, "ScatterMax", scatter_op::UpdateOp::MAX);
172 scatter_op::UpdateOp::ASSIGN);
Dscatter_nd_op.h40 enum class UpdateOp { ASSIGN, ADD, SUB, MIN, MAX }; enum
48 scatter_nd_op::UpdateOp op, int IXDIM>
67 scatter_nd_op::UpdateOp Op>
Dresource_variable_ops.cc837 template <typename Device, typename T, typename Index, scatter_op::UpdateOp op>
954 scatter_op::UpdateOp::ADD); \
956 scatter_op::UpdateOp::SUB); \
958 scatter_op::UpdateOp::MUL); \
960 scatter_op::UpdateOp::DIV); \
962 scatter_op::UpdateOp::ASSIGN);
965 scatter_op::UpdateOp::MIN); \
967 scatter_op::UpdateOp::MAX);
978 scatter_op::UpdateOp::ASSIGN);
980 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/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,
/external/llvm-project/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,
1758 const llvm::function_ref<RValue(RValue)> &UpdateOp, in EmitAtomicUpdateValue() argument
1801 RValue NewRVal = UpdateOp(UpRVal); in EmitAtomicUpdateValue()
1812 llvm::AtomicOrdering AO, const llvm::function_ref<RValue(RValue)> &UpdateOp, in EmitAtomicUpdateLibcall() argument
1831 EmitAtomicUpdateValue(CGF, *this, OldRVal, UpdateOp, DesiredAddr); in EmitAtomicUpdateLibcall()
1841 llvm::AtomicOrdering AO, const llvm::function_ref<RValue(RValue)> &UpdateOp, in EmitAtomicUpdateOp() argument
1863 EmitAtomicUpdateValue(CGF, *this, OldRVal, UpdateOp, NewAtomicAddr); in EmitAtomicUpdateOp()
1953 llvm::AtomicOrdering AO, const llvm::function_ref<RValue(RValue)> &UpdateOp, in EmitAtomicUpdate() argument
[all …]
DCodeGenFunction.h3675 const llvm::function_ref<RValue(RValue)> &UpdateOp,
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view_internal_test.cc79 TEST(MutableNodeViewDiffTest, UpdateOp) { in TEST() argument
94 UpdateOp(&diff, "RandomOp"); in TEST()
98 UpdateOp(&diff, kNodeOp); in TEST()
878 TEST(MutationNewNodeTest, UpdateOp) { in TEST() argument
889 UpdateOp(&new_node, "Identity"); in TEST()
891 UpdateOp(&new_node, ""); in TEST()
Dgraph_view_internal.h418 inline bool UpdateOp(NodeViewDiff<GraphViewT>* diff, absl::string_view op) { in UpdateOp() function
805 inline void UpdateOp(NewNode<GraphViewT>* new_node, absl::string_view op) { in UpdateOp() function
Dgraph_view.cc309 return internal::UpdateOp(diff, op); in UpdateNodeOp()
315 internal::UpdateOp(&new_nodes_[node.index_], op); in UpdateNodeOp()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dcsr_sparse_matrix_to_dense_op.cc200 functor::DoScatterNd<Device, T, int64, scatter_nd_op::UpdateOp::ASSIGN>( in Compute()
/external/llvm-project/mlir/lib/Dialect/OpenACC/IR/
DOpenACC.cpp745 static LogicalResult verify(acc::UpdateOp updateOp) { in verify()
/external/llvm-project/flang/lib/Lower/
DOpenACC.cpp884 auto updateOp = createSimpleOp<mlir::acc::UpdateOp>( in genACCUpdateOp()