Searched refs:grad_batch (Results 1 – 1 of 1) sorted by relevance
659 const int64 grad_batch = grad_shape.dim_size(0); in Compile() local663 OP_REQUIRES(ctx, batch == grad_batch, in Compile()666 batch, " vs. ", grad_batch, ")")); in Compile()