Searched defs:momentum_shape (Results 1 – 1 of 1) sorted by relevance
147 TensorShape momentum_shape = ctx->InputShape(4); in Compile() local205 TensorShape momentum_shape = ctx->InputShape(4); in Compile() local662 TensorShape momentum_shape = ctx->InputShape("momentum"); in Compile() local