Searched refs:const_zero (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | lamb_apply_weight_assign.py | 36 const_zero = graph_builder.value(dtype, 0)
42 g_norm_greater_res = graph_builder.emit('Greater', [g_norm, const_zero])
54 w_norm_greater_res = graph_builder.emit('Greater', [w_norm, const_zero])
|
| D | relu.py | 25 const_zero = graph_builder.value(input_x.dtype, 0)
26 ge_result = graph_builder.emit('Greater', [input_x, const_zero])
|
| D | relu_grad.py | 27 const_zero = graph_builder.value(input_y.dtype, 0)
28 ge_result = graph_builder.emit('Greater', [input_y, const_zero])
|
| D | sigmoid_cross_entropy_with_logits.py | 31 const_zero = graph_builder.value(logits.dtype, 0.0)
32 max_logits = graph_builder.emit('Maximum', [logits, const_zero])
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | convert.cc | 173 auto const_zero = std::make_shared<Constant>("const/npu_runconfig/zero"); in InitLoopVar() local
175 …(void)const_zero->set_attr_value(GeTensor(desc, reinterpret_cast<uint8_t *>(&value), sizeof(int64_… in InitLoopVar()
180 (void)const_zero->update_output_desc_y(desc); in InitLoopVar()
189 (void)assign_zero->set_input_ref(*var_zero).set_input_value(*const_zero); in InitLoopVar()
202 init_ops_.push_back(const_zero); in InitLoopVar()
|