Searched defs:activation_shape (Results 1 – 3 of 3) sorted by relevance
41 const RuntimeShape& activation_shape, float* activation, in GruCell()
55 const RuntimeShape activation_shape = GetTensorShape(activation); in GruImpl() local
411 const Shape activation_shape = activation->shape(); in HandleBatchNormGrad() local