Searched defs:num_input (Results 1 – 9 of 9) sorted by relevance
65 int num_input = model->GetArray(curr_op->inputs[LstmCellOperator::DATA_INPUT]) in Run() local
63 int num_input = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run() local
58 const uint32_t num_input = src.shape()[0]; in qlstm_layer_normalization() local
48 unsigned num_input; // number of inputs member
47 … std::pair<int64_t, int64_t> compute_mean_variance(int64_t sum, int64_t sum_sq, uint32_t num_input) in compute_mean_variance()
1909 int num_input = n->num_inputs(); in MakeIdentityNodesForArgsToLift() local
2042 int num_input; in TEST_P() member
2990 const int num_input = TrtTensorDimsNumElements(GetTestDims(dims_array)); in TEST_P() local
1357 int num_input; in HybridConv() local