Searched defs:alpha_shape (Results 1 – 3 of 3) sorted by relevance
30 const T* input_data, const RuntimeShape& alpha_shape, const T* alpha_data, in BroadcastPrelu4DSlow()77 const T* input_data, const RuntimeShape& alpha_shape, in Prelu()
39 TensorShape alpha_shape = ctx->InputShape(1); in Compile() local84 TensorShape alpha_shape = ctx->InputShape(1); in Compile() local1044 TensorShape alpha_shape = ctx->InputShape(3); in CheckScalarParams() local
7751 const float* input_data, const RuntimeShape& alpha_shape, in BroadcastPReluDispatch()