Searched refs:accum_val_dims (Results 1 – 1 of 1) sorted by relevance
125 int64 accum_val_dims = accum_val_->dims(); in ValidateShape() local126 if (accum_val_dims != grad_val_dims) { in ValidateShape()128 accum_val_dims, ", got ", grad_val_dims); in ValidateShape()130 for (int64 i = 1; i < accum_val_dims; i++) { in ValidateShape()441 int64 accum_val_dims = accum_val_->dims(); in ReturnShapeTensor() local444 ctx, ctx->allocate_output(2, {accum_val_dims}, &shape_tensor)); in ReturnShapeTensor()451 for (int64 i = 1; i < accum_val_dims; i++) { in ReturnShapeTensor()