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