Searched defs:grad_shape (Results 1 – 4 of 4) sorted by relevance
85 TensorShape grad_shape = ctx->InputShape(3); in Compile() local143 TensorShape grad_shape = ctx->InputShape(3); in Compile() local182 TensorShape grad_shape = ctx->InputShape(9); in Compile() local284 TensorShape grad_shape = ctx->InputShape(7); in Compile() local362 TensorShape grad_shape = ctx->InputShape(3); in CompileFtrl() local
386 TensorShape grad_shape = ctx->InputShape(0); in Compile() local
197 MklShape input_shape, grad_shape, output_shape; member
1376 grad_shape=None, argument