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