Searched defs:input_var (Results 1 – 13 of 13) sorted by relevance
67 auto input_var = cnode->input(1); in GetApplyMomentumFormat() local
80 auto input_var = in_tensors_.at(2); in DoExecute() local
65 auto *input_var = in_tensors_.at(4); in DoExecute() local
89 auto input_var = in_tensors_.at(kNumInputDim_2); in DoExecute() local
80 auto *input_var = in_tensors_.at(kNumInputDim_4); in DoExecute() local
37 Tensor *input_var = ctx.Input(0); in Compute() local
41 Tensor *input_var = ctx.Input(0); in Compute() local
170 auto input_var = reinterpret_cast<T *>(ctx.Input(2)->GetData()); in LayerNormGradGradCompute() local
64 bool calc_inv_std(DATA_T *input_var, DATA_T *inv_std, size_t mean_num) { in calc_inv_std()182 auto input_var = static_cast<DATA_T *>(inputs[2]->device_ptr()); in LaunchKernel() local
220 for (const auto &input_var : input_vars_) { in CreateAdamApplyOneNode() local
3258 nir_variable *input_var; in get_matching_input() local3686 should_add_varying_match_record(nir_variable *const input_var, in should_add_varying_match_record()3807 nir_variable *const input_var = in assign_initial_varying_locations() local3903 nir_variable *const input_var = in assign_initial_varying_locations() local
1081 replace_duplicate_input(nir_shader *shader, nir_variable *input_var, in replace_duplicate_input()
13131 static const GLchar* input_var = "in Data dataARRAY;\n"; in test() local