Searched defs:alpha_shape (Results 1 – 2 of 2) sorted by relevance
39 TensorShape alpha_shape = ctx->InputShape(1); in Compile() local84 TensorShape alpha_shape = ctx->InputShape(1); in Compile() local945 TensorShape alpha_shape = ctx->InputShape(3); in CheckScalarParams() local
4603 const RuntimeShape& alpha_shape, in BroadcastPrelu4DSlow()