File tree Expand file tree Collapse file tree 3 files changed +15
-1
lines changed
fluid/pir/dialect/operator Expand file tree Collapse file tree 3 files changed +15
-1
lines changed Original file line number Diff line number Diff line change 163163 args : (Tensor[] x, bool[] share_dims_and_dtype = {})
164164 output : Tensor[](out){x.size()}, Tensor[](xout){x.size()}
165165
166+ - op : share_data
167+ args : (Tensor x)
168+ output : Tensor(out)
169+ kernel :
170+ func : share_data
171+ param : [x]
172+
166173- op : write_to_array
167174 args : (Tensor i, Tensor x)
168175 output : Tensor[](out)
Original file line number Diff line number Diff line change @@ -34,7 +34,8 @@ const std::unordered_set<std::string> LegacyOpList = {
3434 " pd_op.c_reduce_sum_" ,
3535 " pd_op.c_allreduce_max_" ,
3636 " pd_op.c_allgather" ,
37- " pd_op.seed" };
37+ " pd_op.seed" ,
38+ " pd_op.share_data" };
3839
3940enum class AttrType {
4041 UNDEFINED = 0 ,
Original file line number Diff line number Diff line change 31913191 data_type : int64_t
31923192 tensors_name : StepsTensorList
31933193
3194+ - op : share_data
3195+ inputs :
3196+ x : X
3197+ outputs :
3198+ out : Out
3199+
31943200- op : sigmoid_cross_entropy_with_logits
31953201 backward : sigmoid_cross_entropy_with_logits_grad
31963202 inputs :
You can’t perform that action at this time.
0 commit comments