Searched defs:gradient_shape (Results 1 – 1 of 1) sorted by relevance
199 const TensorShape gradient_shape = ctx->InputShape(0); in Compile() local291 const TensorShape gradient_shape = ctx->InputShape(0); in Compile() local