Searched defs:num_input (Results 1 – 12 of 12) sorted by relevance
67 int num_input = model->GetArray(curr_op->inputs[LstmCellOperator::DATA_INPUT]) in Run() local
65 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()
126 const int num_input = grad_input.size(0) * grad_input.stride(0); in conv_depthwise3d_cuda_backward_input_kernel() local
1558 int num_input = input_ty.getNumElements(); in ComputeSliceK() local
787 void VerifyTensorRankAndDType(TFE_TensorHandle** tensors, int num_input, in VerifyTensorRankAndDType()
1867 int num_input = n->num_inputs(); in MakeIdentityNodesForArgsToLift() local
2100 int num_input; in TEST_P() member
3651 const int64_t num_input = DimsAdapter(dims_array).Volume(); in TEST_P() local
1028 int num_input; in HybridConv() local