Searched defs:grad_shape (Results 1 – 3 of 3) sorted by relevance
141 TensorShape grad_shape = ctx->InputShape(3); in Compile() local199 TensorShape grad_shape = ctx->InputShape(3); in Compile() local255 TensorShape grad_shape = ctx->InputShape(3); in Compile() local304 TensorShape grad_shape = ctx->InputShape(5); in Compile() local354 TensorShape grad_shape = ctx->InputShape(3); in Compile() local424 TensorShape grad_shape = ctx->InputShape(9); in Compile() local508 TensorShape grad_shape = ctx->InputShape(8); in Compile() local595 TensorShape grad_shape = ctx->InputShape("grad"); in Compile() local694 TensorShape grad_shape = ctx->InputShape(3); in CompileFtrl() local828 TensorShape grad_shape = ctx->InputShape(6); in Compile() local[all …]
619 TensorShape grad_shape = ctx->InputShape(0); in Compile() local
1392 grad_shape=None, argument