Searched defs:input_element (Results 1 – 8 of 8) sorted by relevance
61 …auto input_element = std::accumulate(input_shape_vec.begin(), input_shape_vec.end(), static_cast<i… in InferShape() local75 …auto input_element = std::accumulate(input_shape_vec.begin(), input_shape_vec.end(), static_cast<i… in InferShape() local
55 int input_element = inputs[0]->size() / sizeof(T); in LaunchKernel() local
169 int64_t input_element{ctx.Input(kIndex0)->NumElements()}; in FFTBaseCompute() local
207 int64_t input_element{ctx.Input(kIndex0)->NumElements()}; in RFFTGradCompute() local
225 int64_t input_element{ctx.Input(kIndex0)->NumElements()}; in IRFFTCompute() local
232 int64_t input_element{ctx.Input(kIndex0)->NumElements()}; in IRFFTGradCompute() local
538 HI_U32 input_element = static_cast<HI_U32>((*inputs)[j].ElementNum()); in FillData() local
301 char **input_element; /* list of fallbacks */ member