Searched refs:num_input (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | random_choice_op.cc | 26 uint32_t num_input = ops_.front()->NumInput(); in NumInput() local
29 if (num_input != cur_num && cur_num > 0) { in NumInput()
34 return num_input; in NumInput()
|
| /third_party/mesa3d/src/gallium/auxiliary/tgsi/ |
| D | tgsi_aa_point.c | 48 unsigned num_input; // number of inputs member
73 ts->num_input++; in aa_decl()
115 texIn = ts->num_input++; in aa_prolog()
303 transform.num_input = 0; in tgsi_add_aa_point()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | replace_addn_fusion.cc | 36 int64_t num_input = AnfAlgo::GetNodeAttr<int64_t>(node, "n"); in Process() local
37 if (num_input == kAddNInputNum) { in Process()
|
| /third_party/mindspore/tests/ |
| D | ops_common.py | 246 def get_loss_fun(construct_net, num_input, output_index): argument
248 f = getattr(net, 'construct%d' % num_input)
411 def get_output_wrap(network, num_input, num_output, output_is_tuple=0): argument
413 f = getattr(net, 'construct%d' % num_input)
|
| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | block_util.py | 80 def get_output_cell(network, num_input, output_index, training=True): argument
81 _ = num_input
|
| /third_party/mindspore/third_party/proto/caffe/ |
| D | caffe.proto | 744 // vector indices with dimension num_input. Hence num_input should be
|