Searched defs:num_output (Results 1 – 11 of 11) sorted by relevance
52 auto num_output = api::GetValue<int64_t>(primitive->GetAttr(kNumOutput)); in Check() local
37 int64_t num_output = static_cast<int64_t>(innerProductParam.num_output()); in Parse() local
39 uint32_t num_output = pass_through_param.num_output(); in Parse() local
44 uint32_t num_output = static_cast<size_t>(api::GetValue<int64_t>(prim->GetAttr(kNumOutput))); in SetNumOutput() local
59 int64_t num_output = 0; in Infer() local
40 uint32_t num_output = ops_.front()->NumOutput(); in NumOutput() local
256 int64_t num_output = ctx.Output(0)->NumElements(); in SliceCompute() local
382 const size_t &num_output = ctx.GetOutputsSize(); in ParseKernelParam() local
90 size_t num_output = 0; in AddInnerOp() local
590 optional uint32 num_output = 1; // The number of outputs for the layer field742 optional uint32 num_output = 1; // The number of outputs for the layer field842 optional uint32 num_output = 1; // The number of outputs for the layer field978 optional uint32 num_output = 1 [default = 0]; field1364 optional uint32 num_output = 3; // The number of outputs for the layer field
344 def __init__(self, network, num_output, output_is_tuple): argument411 def get_output_wrap(network, num_input, num_output, output_is_tuple=0): argument