Home
last modified time | relevance | path

Searched defs:num_input (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dreplace_addn_fusion.cc36 int64_t num_input = AnfAlgo::GetNodeAttr<int64_t>(node, "n"); in Process() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/
Drandom_choice_op.cc26 uint32_t num_input = ops_.front()->NumInput(); in NumInput() local
/third_party/mesa3d/src/gallium/auxiliary/tgsi/
Dtgsi_aa_point.c48 unsigned num_input; // number of inputs member
/third_party/mindspore/tests/
Dops_common.py246 def get_loss_fun(construct_net, num_input, output_index): argument
411 def get_output_wrap(network, num_input, num_output, output_is_tuple=0): argument
/third_party/mindspore/tests/mindspore_test_framework/utils/
Dblock_util.py80 def get_output_cell(network, num_input, output_index, training=True): argument