Searched defs:lr_shape (Results 1 – 1 of 1) sorted by relevance
136 TensorShape lr_shape = ctx->InputShape(2); in Compile() local194 TensorShape lr_shape = ctx->InputShape(2); in Compile() local252 TensorShape lr_shape = ctx->InputShape(2); in Compile() local301 TensorShape lr_shape = ctx->InputShape(2); in Compile() local355 TensorShape lr_shape = ctx->InputShape(2); in Compile() local418 TensorShape lr_shape = ctx->InputShape(4); in Compile() local484 TensorShape lr_shape = ctx->InputShape(5); in Compile() local579 TensorShape lr_shape = ctx->InputShape(4); in Compile() local654 TensorShape lr_shape = ctx->InputShape("lr"); in Compile() local771 TensorShape lr_shape = ctx->InputShape(4); in CompileFtrl() local[all …]